Update interim.py

interim.py

@@ -247,15 +247,41 @@ class DocumentRAG:
 if "rag_system" not in st.session_state:
     st.session_state.rag_system = DocumentRAG()
 
-# Streamlit UI
-st.title("Document Analyzer and Podcast Generator")
-
-# Fetch the API key status
-if "OPENAI_API_KEY" not in os.environ or not os.getenv("OPENAI_API_KEY"):
-    st.error("The 'OPENAI_API_KEY' environment variable is not set. Please configure it in your hosting environment.")
+# Sidebar
+with st.sidebar:
+    st.title("About")
+    st.markdown(
+        """
+        This app is inspired by the [RAG_HW HuggingFace Space](https://huggingface.co/spaces/wint543/RAG_HW).  
+        It allows users to upload documents, generate summaries, ask questions, and create podcasts.
+        """
+    )
+    st.markdown("### Steps:")
+    st.markdown("1. Upload documents.")
+    st.markdown("2. Generate summaries.")
+    st.markdown("3. Ask questions.")
+    st.markdown("4. Create podcasts.")
 
-# File upload
-st.subheader("Step 1: Upload Documents")
+# Streamlit UI
+# Sidebar
+#with st.sidebar:
+    #st.title("About")
+    #st.markdown(
+        #"""
+        #This app is inspired by the [RAG_HW HuggingFace Space](https://huggingface.co/spaces/wint543/RAG_HW).  
+        #It allows users to:
+        #1. Upload and process documents
+        #2. Generate summaries
+        #3. Ask questions
+        #4. Create podcasts
+        #"""
+    #)
+
+# Main App
+st.title("Document Analyzer & Podcast Generator")
+
+# Step 1: Upload and Process Documents
+st.subheader("Step 1: Upload and Process Documents")
 uploaded_files = st.file_uploader("Upload files (PDF, TXT, CSV)", accept_multiple_files=True)
 
 if st.button("Process Documents"):
@@ -269,25 +295,66 @@ if st.button("Process Documents"):
     else:
         st.warning("No files uploaded.")
 
-# Document Q&A
-st.subheader("Step 2: Ask Questions")
+# 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, 
+    horizontal=True, 
+    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
+    )
+
+# Step 3: Ask Questions
+st.subheader("Step 3: Ask Questions")
+st.write("Select Q&A Language:")
+qa_language_options = ["English", "Hindi", "Spanish", "French", "German", "Chinese", "Japanese"]
+qa_language = st.radio(
+    "", 
+    qa_language_options, 
+    horizontal=True, 
+    key="qa_language"
+)
+
 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)
 else:
-    st.info("Please process documents before asking questions.")
+    st.info("Please process documents first to enable Q&A.")
+
+# Step 4: Generate Podcast
+st.subheader("Step 4: Generate Podcast")
+st.write("Select Podcast Language:")
+podcast_language_options = ["English", "Hindi", "Spanish", "French", "German", "Chinese", "Japanese"]
+podcast_language = st.radio(
+    "", 
+    podcast_language_options, 
+    horizontal=True, 
+    key="podcast_language"
+)
 
-# Podcast Generation
-st.subheader("Step 3: Generate Podcast")
 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")
@@ -295,4 +362,4 @@ if st.session_state.rag_system.document_summary:
         else:
             st.error(script)
 else:
-    st.info("Please process documents to generate a podcast.")
+    st.info("Please process documents and generate summaries before creating a podcast.")