Update interim.py

interim.py

@@ -85,9 +85,9 @@ class DocumentRAG:
             )
             documents = text_splitter.split_documents(documents)
 
-            # Combine text for summary
-            combined_text = " ".join([doc.page_content for doc in documents])
-            self.document_summary = self.generate_summary(combined_text)
+            # Combine text for later summary generation
+            self.document_text = " ".join([doc.page_content for doc in documents])  # Store for later use
+
 
             # Create embeddings and initialize retrieval chain
             embeddings = OpenAIEmbeddings(api_key=self.api_key)
@@ -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):
+        """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):
+        """Generate a podcast script and audio based on doc summary in the specified language."""
         if not self.document_summary:
             return "Please process documents before generating a podcast.", None
 
@@ -142,7 +142,7 @@ 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": "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:
                     1. Clearly label the dialogue as 'Host 1:' and 'Host 2:'
                     2. Keep the content engaging and insightful.
@@ -201,31 +201,13 @@ 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):
+        """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.
+            Instruction: Respond in {language}. 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."
             """
             query = f"{preface}\nQuery: {question}"
@@ -295,12 +277,11 @@ if st.button("Process Documents"):
     else:
         st.warning("No files uploaded.")
 
+    
 # Step 2: Generate Summaries
 st.subheader("Step 2: Generate Summaries")
 st.write("Select Summary Language:")
 summary_language_options = ["English", "Hindi", "Spanish", "French", "German", "Chinese", "Japanese"]
-cols = st.columns(len(summary_language_options))
-# Default selected option
 summary_language = st.radio(
     "", 
     summary_language_options, 
@@ -308,15 +289,14 @@ summary_language = st.radio(
     key="summary_language"
 )
 
-if st.session_state.rag_system.document_summary:
-    st.text_area("Document Summary", st.session_state.rag_system.document_summary, height=200)
-else:
-    st.info("Please process documents first to generate summaries.")
 if st.button("Generate Summary"):
-    st.session_state.rag_system.document_summary = st.session_state.rag_system.generate_summary(
-        st.session_state.rag_system.document_summary, 
-        summary_language
-    )
+    if hasattr(st.session_state.rag_system, "document_text") and st.session_state.rag_system.document_text:
+        summary = st.session_state.rag_system.generate_summary(st.session_state.rag_system.document_text, summary_language)
+        st.session_state.rag_system.document_summary = summary
+        st.text_area("Document Summary", summary, height=200)
+    else:
+        st.info("Please process documents first to generate summaries.")
+
 
 # Step 3: Ask Questions
 st.subheader("Step 3: Ask Questions")
@@ -362,4 +342,4 @@ if st.session_state.rag_system.document_summary:
         else:
             st.error(script)
 else:
-    st.info("Please process documents and generate summaries before creating a podcast.")
+    st.info("Please process documents and generate summaries before creating a podcast.")