Update app.py

app.py

@@ -262,27 +262,6 @@ with st.sidebar:
     st.markdown("3. Ask questions.")
     st.markdown("4. Create podcasts.")
 
-
-# Sidebar: Language Selection
-with st.sidebar:
-    st.title("Settings")
-    summary_language = st.radio(
-        "Select Summary Language:", 
-        ["English", "Spanish", "French", "German", "Chinese"], 
-        index=0
-    )
-    qa_language = st.radio(
-        "Select Q&A Language:", 
-        ["English", "Spanish", "French", "German", "Chinese"], 
-        index=0
-    )
-    podcast_language = st.radio(
-        "Select Podcast Language:", 
-        ["English", "Spanish", "French", "German", "Chinese"], 
-        index=0
-    )
-
-
 # Streamlit UI
 # Sidebar
 #with st.sidebar:
@@ -318,19 +297,34 @@ if st.button("Process Documents"):
 
 # Step 2: Generate Summaries
 st.subheader("Step 2: Generate Summaries")
+summary_language = st.radio(
+    "Select Summary Language:",
+    ["English", "Spanish", "French", "German", "Chinese"],
+    index=0
+)
 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
+    )
 
 # Step 3: Ask Questions
 st.subheader("Step 3: Ask Questions")
+qa_language = st.radio(
+    "Select Q&A Language:",
+    ["English", "Spanish", "French", "German", "Chinese"],
+    index=0
+)
 if st.session_state.rag_system.qa_chain:
     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)
+        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)
@@ -339,9 +333,14 @@ else:
 
 # Step 4: Generate Podcast
 st.subheader("Step 4: Generate Podcast")
+podcast_language = st.radio(
+    "Select Podcast Language:",
+    ["English", "Spanish", "French", "German", "Chinese"],
+    index=0
+)
 if st.session_state.rag_system.document_summary:
     if st.button("Generate Podcast"):
-        script, audio_path = st.session_state.rag_system.create_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")
@@ -349,4 +348,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.")