Update app.py

app.py

@@ -109,8 +109,8 @@ class DocumentRAG:
         except Exception as e:
             return f"Error processing documents: {str(e)}"
 
-    def generate_summary(self, text):
-        """Generate a summary of the provided text."""
+    def generate_summary(self, text, language="English"):
+        """Generate a summary of the provided text in the specified language."""
         if not self.api_key:
             return "API Key not set. Please set it in the environment variables."
         try:
@@ -118,7 +118,7 @@ class DocumentRAG:
             response = client.chat.completions.create(
                 model="gpt-4",
                 messages=[
-                    {"role": "system", "content": "Summarize the document content concisely and provide 3-5 key points for discussion."},
+                    {"role": "system", "content": f"Summarize the document content concisely in {language}. Provide 3-5 key points for discussion."},
                     {"role": "user", "content": text[:4000]}
                 ],
                 temperature=0.3
@@ -127,8 +127,8 @@ class DocumentRAG:
         except Exception as e:
             return f"Error generating summary: {str(e)}"
 
-    def create_podcast(self):
-        """Generate a podcast script and audio based on the document summary."""
+    def create_podcast(self, language="English"):
+        """Generate a podcast script and audio based on the document summary in the specified language."""
         if not self.document_summary:
             return "Please process documents before generating a podcast.", None
 
@@ -142,8 +142,8 @@ class DocumentRAG:
             script_response = client.chat.completions.create(
                 model="gpt-4",
                 messages=[
-                    {"role": "system", "content": "You are a professional podcast producer. Create a natural dialogue based on the provided document summary."},
-                    {"role": "user", "content": f"""Based on the following document summary, create a 1-2 minute podcast script:
+                    {"role": "system", "content": f"You are a professional podcast producer. Create a natural dialogue in {language} based on the provided document summary."},
+                    {"role": "user", "content": f"""Based on the following document summary, create a 1-2 minute podcast script in {language}:
                     1. Clearly label the dialogue as 'Host 1:' and 'Host 2:'
                     2. Keep the content engaging and insightful.
                     3. Use conversational language suitable for a podcast.
@@ -201,32 +201,15 @@ class DocumentRAG:
         except Exception as e:
             return f"Error generating podcast: {str(e)}", None
 
-    def generate_summary(self, text):
-        """Generate a summary of the provided text."""
-        if not self.api_key:
-            return "API Key not set. Please set it in the environment variables."
-        try:
-            client = OpenAI(api_key=self.api_key)
-            response = client.chat.completions.create(
-                model="gpt-4",
-                messages=[
-                    {"role": "system", "content": "Summarize the document content concisely and provide 3-5 key points for discussion."},
-                    {"role": "user", "content": text[:4000]}
-                ],
-                temperature=0.3
-            )
-            return response.choices[0].message.content
-        except Exception as e:
-            return f"Error generating summary: {str(e)}"
 
-    def handle_query(self, question, history):
-        """Handle user queries."""
+    def handle_query(self, question, history, language="English"):
+        """Handle user queries in the specified language."""
         if not self.qa_chain:
             return history + [("System", "Please process the documents first.")]
         try:
-            preface = """
-            Instruction: Respond in English. Be professional and concise, keeping the response under 300 words.
-            If you cannot provide an answer, say: "I am not sure about this question. Please try asking something else."
+            preface = f"""
+            Instruction: Respond in {language}. Be professional and concise, keeping the response under 300 words.
+            If you cannot provide an answer, say in the specified language: "I am not sure about this question. Please try asking something else."
             """
             query = f"{preface}\nQuery: {question}"
 
@@ -308,7 +291,7 @@ summary_language = st.radio(
     key="summary_language"
 )
 
-# Confirm selected language
+# Debugging: Confirm selected language
 st.write(f"Selected Summary Language: {summary_language}")
 
 if st.session_state.rag_system.document_summary:
@@ -316,10 +299,14 @@ if st.session_state.rag_system.document_summary:
 else:
     st.info("Please process documents first to generate summaries.")
 if st.button("Generate Summary"):
+    # Combine all document text
+    raw_text = " ".join([doc.page_content for doc in st.session_state.rag_system.document_store.documents])
     st.session_state.rag_system.document_summary = st.session_state.rag_system.generate_summary(
-        st.session_state.rag_system.document_summary, 
+        raw_text,  # Pass the raw document content
         summary_language
     )
+# Debugging: Confirm selected language
+st.write(f"Selected Summary Language: {summary_language}")
 
 # Step 3: Ask Questions
 st.subheader("Step 3: Ask Questions")
@@ -333,14 +320,25 @@ qa_language = st.radio(
 )
 
 if st.session_state.rag_system.qa_chain:
-    history = []
+    # Retain chat history in session state
+    if "chat_history" not in st.session_state:
+        st.session_state.chat_history = []
+
     user_question = st.text_input("Ask a question:")
+
     if st.button("Submit Question"):
-        # Handle the user query
-        history = st.session_state.rag_system.handle_query(user_question, history, qa_language)
-        for question, answer in history:
-            st.chat_message("user").write(question)
-            st.chat_message("assistant").write(answer)
+        if user_question.strip():
+            # Handle the user query in the selected language
+            st.session_state.chat_history = st.session_state.rag_system.handle_query(
+                user_question,
+                st.session_state.chat_history,
+                qa_language
+            )
+            for question, answer in st.session_state.chat_history:
+                st.chat_message("user").write(question)
+                st.chat_message("assistant").write(answer)
+        else:
+            st.warning("Please enter a valid question.")
 else:
     st.info("Please process documents first to enable Q&A.")
 
@@ -354,15 +352,26 @@ podcast_language = st.radio(
     horizontal=True, 
     key="podcast_language"
 )
-
-if st.session_state.rag_system.document_summary:
+if st.session_state.rag_system.document_store:
+    podcast_language = st.session_state.get("podcast_language", "English")  # Get the selected podcast language
     if st.button("Generate Podcast"):
-        script, audio_path = st.session_state.rag_system.create_podcast(podcast_language)
-        if audio_path:
-            st.text_area("Generated Podcast Script", script, height=200)
-            st.audio(audio_path, format="audio/mp3")
-            st.success("Podcast generated successfully! You can listen to it above.")
+        # Fetch the raw document content (combine all chunks if needed)
+        raw_document_text = " ".join(
+            [doc.page_content for doc in st.session_state.rag_system.document_store.documents]
+        )
+
+        if raw_document_text:
+            # Generate the podcast script and audio
+            script, audio_path = st.session_state.rag_system.create_podcast(raw_document_text, language=podcast_language)
+
+            # Check if audio generation was successful
+            if audio_path:
+                st.text_area("Generated Podcast Script", script, height=200)
+                st.audio(audio_path, format="audio/mp3")
+                st.success("Podcast generated successfully! You can listen to it above.")
+            else:
+                st.error(f"Failed to generate podcast: {script}")
         else:
-            st.error(script)
+            st.warning("No document content found. Please process documents before generating a podcast.")
 else:
-    st.info("Please process documents and generate summaries before creating a podcast.")
+    st.info("Please process documents before attempting to generate a podcast.")