Update app.py

app.py

@@ -4,7 +4,7 @@ from components.sidebar import render_sidebar
 from components.paper_list import render_paper_list
 from components.notebook_view import render_notebook
 from components.graph_view import render_graph
-from orchestrator.gemini import gemini_generate
+from orchestrator.gemini import gemini_generate, gemini_generate_code
 
 def main():
     st.set_page_config(
@@ -13,10 +13,10 @@ def main():
         initial_sidebar_state="expanded"
     )
 
-    # Sidebar: user query, num results, theme, gemini prompt, etc.
-    query, num_results, theme, search_clicked, gemini_prompt = render_sidebar()
+    # Sidebar: (note new returned variables)
+    (query, num_results, theme, search_clicked, gemini_prompt,
+     mcp_codegen_prompt, codegen_clicked) = render_sidebar()
 
-    # Optional theme
     if theme == "Dark":
         st.markdown(
             """
@@ -28,7 +28,34 @@ def main():
             unsafe_allow_html=True,
         )
 
-    # Search and display papers
+    # -- Gemini Q&A --
+    if gemini_prompt:
+        st.header("💡 Gemini Research Q&A")
+        with st.spinner("Gemini is thinking..."):
+            answer = gemini_generate(gemini_prompt)
+        st.success(answer)
+
+    # -- MCP Code Generation "WOW" Feature --
+    if mcp_codegen_prompt and codegen_clicked:
+        st.header("🛠️ Gemini MCP Server Code Generation")
+        with st.spinner("Gemini is coding your MCP server..."):
+            # You can change the system instruction for advanced results:
+            system_instruction = (
+                "You are an expert in Model Context Protocol (MCP) server development. "
+                "Generate clean, production-ready Python code for an MCP server as described by the user. "
+                "Use best practices and include all necessary imports and comments."
+            )
+            code_result = gemini_generate_code(system_instruction, mcp_codegen_prompt)
+        st.code(code_result, language="python")
+        # Add download and copy to clipboard
+        st.download_button("Download code as mcp_server.py", code_result, file_name="mcp_server.py", mime="text/x-python")
+        try:
+            import streamlit_copy_to_clipboard as stc
+            stc.copy_to_clipboard(code_result, "Copy to clipboard")
+        except ImportError:
+            st.info("Install `streamlit-copy-to-clipboard` for copy button.")
+
+    # -- Search and Display Papers --
     if search_clicked and query:
         dispatcher = Dispatcher()
         with st.spinner("Searching MCP servers..."):
@@ -43,19 +70,11 @@ def main():
                     summary = gemini_generate(first_paper["abstract"])
                     st.success(summary)
 
-            # Show notebook and graph
             notebook_cells = dispatcher.get_notebook_cells(first_paper["id"])
             render_notebook(notebook_cells)
 
             graph_data = dispatcher.get_graph(first_paper["id"])
             render_graph(graph_data)
 
-    # Gemini research Q&A, even if user hasn't hit 'Search'
-    if gemini_prompt:
-        st.header("💡 Gemini Research Q&A")
-        with st.spinner("Gemini is thinking..."):
-            answer = gemini_generate(gemini_prompt)
-        st.success(answer)
-
 if __name__ == "__main__":
     main()