Update app.py

app.py

@@ -113,7 +113,7 @@ if "documents" in st.session_state:
         "Identify research gaps",
         "Suggest research ideas",
         "Simulate a debate",
-        "Translate summary",
+        "Translate response",
         "Generate citation"
     ])
 
@@ -134,7 +134,7 @@ if "documents" in st.session_state:
             if task == "Summarize document":
                 chain = create_stuff_documents_chain(llm, summary_prompt)
                 summary = run_chain(chain, {"context": docs})
-                results["summary"] = summary
+                st.session_state["last_agent_output"] = summary
                 st.markdown("### 📝 Summary")
                 st.write(summary)
 
@@ -144,7 +144,7 @@ if "documents" in st.session_state:
                 summary = run_chain(chain1, {"context": docs})
                 chain2 = LLMChain(llm=llm, prompt=gap_prompt)
                 gaps = run_chain(chain2, {"summary": summary})
-                results["gaps"] = gaps
+                st.session_state["last_agent_output"] = gaps
                 st.markdown("### 🔍 Identified Gaps")
                 st.write(gaps)
 
@@ -156,6 +156,7 @@ if "documents" in st.session_state:
                 gaps = run_chain(chain2, {"summary": summary})
                 chain3 = LLMChain(llm=llm, prompt=idea_prompt)
                 ideas = run_chain(chain3, {"gaps": gaps})
+                st.session_state["last_agent_output"] = ideas
                 st.markdown("### 💡 Research Ideas")
                 st.write(ideas)
 
@@ -165,21 +166,25 @@ if "documents" in st.session_state:
                 summary = run_chain(chain, {"context": docs})
                 debate_chain = LLMChain(llm=llm, prompt=debate_prompt)
                 debate = run_chain(debate_chain, {"summary": summary})
+                st.session_state["last_agent_output"] = debate
                 st.markdown("### 🎭 Debate")
                 st.write(debate)
 
             # Translate agent
-            elif task == "Translate summary":
-                chain = create_stuff_documents_chain(llm, summary_prompt)
-                summary = run_chain(chain, {"context": docs})
-                translate_chain = LLMChain(llm=llm, prompt=translate_prompt)
-                translated = run_chain(translate_chain, {"language": user_language, "content": summary})
-                st.markdown(f"### 🌐 Translated Summary ({user_language})")
-                st.write(translated)
+            elif task == "Translate response":
+                if "last_agent_output" in st.session_state:
+                    to_translate = st.session_state["last_agent_output"]
+                    translate_chain = LLMChain(llm=llm, prompt=translate_prompt)
+                    translated = translate_chain.invoke({"language": user_language, "content": to_translate})
+                    st.markdown(f"### 🌐 Translated Response ({user_language})")
+                    st.write(translated)
+                else:
+                    st.warning("Please run another agent first (e.g., Summary, Gap, Debate) to generate something to translate.")
 
             # Citation agent
             elif task == "Generate citation":
                 citation_chain = create_stuff_documents_chain(llm, citation_prompt)
                 citation = run_chain(citation_chain, {"context": docs})
+                st.session_state["last_agent_output"] = citation
                 st.markdown("### 📎 APA Citation")
                 st.code(citation, language="markdown")