Fix long lines and unused imports

app.py

@@ -7,8 +7,6 @@ import os
 import pathlib
 import random
 import sys
-import tempfile
-from typing import List, Union
 
 import httpx
 import huggingface_hub
@@ -27,8 +25,10 @@ import slidedeckai.helpers.file_manager as filem
 from slidedeckai.helpers.chat_helper import ChatMessage, HumanMessage, AIMessage
 from slidedeckai.helpers import chat_helper
 
+
 load_dotenv()
 
+
 class StreamlitChatMessageHistory:
     """Chat message history stored in Streamlit session state."""
 
@@ -39,14 +39,18 @@ class StreamlitChatMessageHistory:
 
     @property
     def messages(self):
+        """Get all chat messages in the history."""
         return st.session_state[self.key]
 
     def add_user_message(self, content: str):
+        """Add a user message to the history."""
         st.session_state[self.key].append(HumanMessage(content))
 
     def add_ai_message(self, content: str):
+        """Add an AI message to the history."""
         st.session_state[self.key].append(AIMessage(content))
 
+
 RUN_IN_OFFLINE_MODE = os.getenv('RUN_IN_OFFLINE_MODE', 'False').lower() == 'true'
 
 
@@ -56,7 +60,6 @@ def _load_strings() -> dict:
     Load various strings to be displayed in the app.
     :return: The dictionary of strings.
     """
-
     with open(GlobalConfig.APP_STRINGS_FILE, 'r', encoding='utf-8') as in_file:
         return json5.loads(in_file.read())
 
@@ -233,8 +236,14 @@ with st.sidebar:
             selected_provider = provider_match.group(1)
         else:
             # If regex doesn't match, try to extract provider from the beginning
-            selected_provider = llm_provider_to_use.split(' ')[0] if ' ' in llm_provider_to_use else llm_provider_to_use
-            logger.warning("Provider regex did not match for: %s, using: %s", llm_provider_to_use, selected_provider)
+            selected_provider = (
+                llm_provider_to_use.split(' ')[0]
+                if ' ' in llm_provider_to_use else llm_provider_to_use
+            )
+            logger.warning(
+                'Provider regex did not match for: %s, using: %s',
+                llm_provider_to_use, selected_provider
+            )
         
         # Validate that the selected provider is valid
         if selected_provider not in GlobalConfig.VALID_PROVIDERS:
@@ -400,11 +409,16 @@ def set_up_chat_ui():
         progress_bar = st.progress(0, 'Preparing to call LLM...')
 
         def progress_callback(current_progress):
-            progress_bar.progress(min(current_progress / gcfg.get_max_output_tokens(llm_provider_to_use), 0.95), text='Streaming content...this might take a while...')
+            progress_bar.progress(
+                min(current_progress / gcfg.get_max_output_tokens(llm_provider_to_use), 0.95),
+                text='Streaming content...this might take a while...'
+            )
 
         try:
             if _is_it_refinement():
-                path = slide_generator.revise(instructions=prompt_text, progress_callback=progress_callback)
+                path = slide_generator.revise(
+                    instructions=prompt_text, progress_callback=progress_callback
+                )
             else:
                 path = slide_generator.generate(progress_callback=progress_callback)
 
@@ -484,43 +498,24 @@ def _is_it_refinement() -> bool:
     return False
 
 
-def _get_user_messages() -> List[str]:
+def _get_user_messages() -> list[str]:
     """
     Get a list of user messages submitted until now from the session state.
 
     :return: The list of user messages.
     """
-
     return [
-        msg.content for msg in st.session_state[CHAT_MESSAGES] if isinstance(msg, chat_helper.HumanMessage)
+        msg.content for msg in st.session_state[CHAT_MESSAGES]
+        if isinstance(msg, chat_helper.HumanMessage)
     ]
 
 
-def _get_last_response() -> str:
-    """
-    Get the last response generated by AI.
-
-    :return: The response text.
-    """
-
-def _display_messages_history(view_messages: st.expander):
-    """
-    Display the history of messages.
-
-    :param view_messages: The list of AI and Human messages.
-    """
-
-    with view_messages:
-        view_messages.json(st.session_state[CHAT_MESSAGES])
-
-
 def _display_download_button(file_path: pathlib.Path):
     """
     Display a download button to download a slide deck.
 
     :param file_path: The path of the .pptx file.
     """
-
     with open(file_path, 'rb') as download_file:
         st.download_button(
             'Download PPTX file ⬇️',