fix: Address PR comments

- Refactor PDF handling to be internal to SlideDeckAI class.

app.py

@@ -386,12 +386,6 @@ def set_up_chat_ui():
                         f' {st.session_state["end_page"]} in {st.session_state["pdf_file"].name}'
                     )
 
-            # Get pdf contents
-            st.session_state[ADDITIONAL_INFO] = filem.get_pdf_contents(
-                st.session_state[PDF_FILE_KEY],
-                (st.session_state['start_page'], st.session_state['end_page'])
-            )
-
         st.chat_message('user').write(prompt_text)
 
         slide_generator = SlideDeckAI(
@@ -399,7 +393,8 @@ def set_up_chat_ui():
             topic=prompt_text,
             api_key=api_key_token.strip(),
             template_idx=list(GlobalConfig.PPTX_TEMPLATE_FILES.keys()).index(pptx_template),
-            additional_info=st.session_state.get(ADDITIONAL_INFO, ''),
+            pdf_path_or_stream=st.session_state.get(PDF_FILE_KEY),
+            pdf_page_range=(st.session_state.get('start_page'), st.session_state.get('end_page')),
         )
 
         progress_bar = st.progress(0, 'Preparing to call LLM...')

src/slidedeckai/core.py

@@ -21,30 +21,30 @@ RUN_IN_OFFLINE_MODE = os.getenv('RUN_IN_OFFLINE_MODE', 'False').lower() == 'true
 
 logger = logging.getLogger(__name__)
 
+from .helpers import file_manager as filem
+
 class SlideDeckAI:
     """
     The main class for generating slide decks.
     """
 
-    def __init__(self, model, topic, api_key=None, pdf_file_path=None, pdf_page_range=None, template_idx=0, additional_info=''):
+    def __init__(self, model, topic, api_key=None, pdf_path_or_stream=None, pdf_page_range=None, template_idx=0):
         """
         Initializes the SlideDeckAI object.
 
         :param model: The name of the LLM model to use.
         :param topic: The topic of the slide deck.
         :param api_key: The API key for the LLM provider.
-        :param pdf_file_path: The path to a PDF file to use as a source for the slide deck.
+        :param pdf_path_or_stream: The path to a PDF file or a file-like object.
         :param pdf_page_range: A tuple representing the page range to use from the PDF file.
         :param template_idx: The index of the PowerPoint template to use.
-        :param additional_info: Additional information to be sent to the LLM, such as text from a PDF.
         """
         self.model = model
         self.topic = topic
         self.api_key = api_key
-        self.pdf_file_path = pdf_file_path
+        self.pdf_path_or_stream = pdf_path_or_stream
         self.pdf_page_range = pdf_page_range
         self.template_idx = template_idx
-        self.additional_info = additional_info
         self.chat_history = ChatMessageHistory()
         self.last_response = None
 
@@ -68,9 +68,13 @@ class SlideDeckAI:
         Generates the initial slide deck.
         :return: The path to the generated .pptx file.
         """
+        additional_info = ''
+        if self.pdf_path_or_stream:
+            additional_info = filem.get_pdf_contents(self.pdf_path_or_stream, self.pdf_page_range)
+
         self.chat_history.add_user_message(self.topic)
         prompt_template = self._get_prompt_template(is_refinement=False)
-        formatted_template = prompt_template.format(question=self.topic, additional_info=self.additional_info)
+        formatted_template = prompt_template.format(question=self.topic, additional_info=additional_info)
 
         provider, llm_name = llm_helper.get_provider_model(self.model, use_ollama=RUN_IN_OFFLINE_MODE)
 
@@ -118,10 +122,14 @@ class SlideDeckAI:
 
         list_of_msgs = [f'{idx + 1}. {msg.content}' for idx, msg in enumerate(self.chat_history.messages) if msg.role == 'user']
 
+        additional_info = ''
+        if self.pdf_path_or_stream:
+            additional_info = filem.get_pdf_contents(self.pdf_path_or_stream, self.pdf_page_range)
+
         formatted_template = prompt_template.format(
             instructions='\n'.join(list_of_msgs),
             previous_content=self.last_response,
-            additional_info=self.additional_info,
+            additional_info=additional_info,
         )
 
         provider, llm_name = llm_helper.get_provider_model(self.model, use_ollama=RUN_IN_OFFLINE_MODE)