Update app.py

app.py

@@ -19,8 +19,10 @@ import extra_streamlit_components as stx
 from streamlit.runtime.scriptrunner import get_script_run_ctx
 import asyncio
 import edge_tts
+import io
+import sys
 
-# 🎯 1. Core Configuration & Setup
+# 1. Core Configuration & Setup
 st.set_page_config(
     page_title="🚲BikeAI🏆 Claude/GPT Research",
     page_icon="🚲🏆",
@@ -34,7 +36,7 @@ st.set_page_config(
 )
 load_dotenv()
 
-# 🔑 2. API Setup & Clients
+# 2. API Setup & Clients
 openai_api_key = os.getenv('OPENAI_API_KEY', "")
 anthropic_key = os.getenv('ANTHROPIC_API_KEY_3', "")
 if 'OPENAI_API_KEY' in st.secrets:
@@ -48,7 +50,7 @@ openai_client = OpenAI(api_key=openai.api_key, organization=os.getenv('OPENAI_OR
 HF_KEY = os.getenv('HF_KEY')
 API_URL = os.getenv('API_URL')
 
-# 📝 3. Session State Management
+# 3. Session State Management
 if 'transcript_history' not in st.session_state:
     st.session_state['transcript_history'] = []
 if 'chat_history' not in st.session_state:
@@ -72,7 +74,7 @@ if 'should_rerun' not in st.session_state:
 if 'old_val' not in st.session_state:
     st.session_state['old_val'] = None
 
-# 🎨 4. Custom CSS
+# 4. Custom CSS
 st.markdown("""
 <style>
     .main { background: linear-gradient(to right, #1a1a1a, #2d2d2d); color: #fff; }
@@ -88,7 +90,7 @@ FILE_EMOJIS = {
     "mp3": "🎵",
 }
 
-# 🧠 5. High-Information Content Extraction
+# 5. High-Information Content Extraction
 def get_high_info_terms(text: str) -> list:
     """Extract high-information terms from text, including key phrases"""
     stop_words = set([
@@ -141,9 +143,8 @@ def get_high_info_terms(text: str) -> list:
     max_terms = 5
     return unique_terms[:max_terms]
 
-# 📁 6. File Operations
+# 6. Filename Generation
 def generate_filename(content, file_type="md"):
-    """Generate filename with meaningful terms"""
     prefix = datetime.now().strftime("%y%m_%H%M") + "_"
     info_terms = get_high_info_terms(content)
     name_text = '_'.join(term.replace(' ', '-') for term in info_terms) if info_terms else 'file'
@@ -155,20 +156,7 @@ def generate_filename(content, file_type="md"):
     filename = f"{prefix}{name_text}.{file_type}"
     return filename
 
-def create_file(prompt, response, file_type="md"):
-    """Create file with intelligent naming"""
-    filename = generate_filename(response.strip() if response.strip() else prompt.strip(), file_type)
-    with open(filename, 'w', encoding='utf-8') as f:
-        f.write(prompt + "\n\n" + response)
-    return filename
-
-def get_download_link(file):
-    """Generate download link for file"""
-    with open(file, "rb") as f:
-        b64 = base64.b64encode(f.read()).decode()
-    return f'<a href="data:file/zip;base64,{b64}" download="{os.path.basename(file)}">📂 Download {os.path.basename(file)}</a>'
-
-# 🔊 7. Audio Processing
+# 7. Audio Processing
 def clean_for_speech(text: str) -> str:
     """Clean text for speech synthesis"""
     text = text.replace("\n", " ")
@@ -214,7 +202,7 @@ def play_and_download_audio(file_path):
         dl_link = f'<a href="data:audio/mpeg;base64,{base64.b64encode(open(file_path,"rb").read()).decode()}" download="{os.path.basename(file_path)}">Download {os.path.basename(file_path)}</a>'
         st.markdown(dl_link, unsafe_allow_html=True)
 
-# 🎬 8. Media Processing
+# 8. Media Processing
 def process_image(image_path, user_prompt):
     """Process image with GPT-4V"""
     with open(image_path, "rb") as imgf:
@@ -271,7 +259,7 @@ def process_video_with_gpt(video_path, prompt):
     )
     return resp.choices[0].message.content
 
-# 🤖 9. AI Model Integration
+# 9. AI Model Integration
 def perform_ai_lookup(q, vocal_summary=True, extended_refs=False, titles_summary=True, full_audio=False):
     """Perform Arxiv search and generate audio summaries"""
     start = time.time()
@@ -318,7 +306,114 @@ def perform_ai_lookup(q, vocal_summary=True, extended_refs=False, titles_summary
 
     elapsed = time.time()-start
     st.write(f"**Total Elapsed:** {elapsed:.2f} s")
-    create_file(q, result, "md")
+
+    # We must provide a filename for the create_file function:
+    # Use generate_filename from 'q' and 'result'
+    filename = generate_filename(result, "md")
+    create_file(filename, q, result, should_save=True)
+
+    # --- Code Interpreter Integration ---
+    # Parse out papers from refs if available
+    # Format assumed:
+    # [Title] Title of Paper
+    # Summary: ...
+    # Link: ...
+    # PDF: ...
+    # separate by "[Title]"
+    papers_raw = refs.strip().split("[Title]")
+    papers = []
+    for p in papers_raw:
+        p = p.strip()
+        if not p:
+            continue
+        lines = p.split("\n")
+        title_line = lines[0].strip() if lines else ""
+        summary_line = ""
+        link_line = ""
+        pdf_line = ""
+        for line in lines[1:]:
+            line = line.strip()
+            if line.startswith("Summary:"):
+                summary_line = line.replace("Summary:", "").strip()
+            elif line.startswith("Link:"):
+                link_line = line.replace("Link:", "").strip()
+            elif line.startswith("PDF:"):
+                pdf_line = line.replace("PDF:", "").strip()
+
+        if title_line and summary_line:
+            papers.append({
+                "title": title_line,
+                "summary": summary_line,
+                "link": link_line,
+                "pdf": pdf_line
+            })
+
+    st.write("## Code Interpreter Options for Each Paper")
+    for i, paper in enumerate(papers):
+        st.write(f"**Paper {i+1}:** {paper['title']}")
+        st.write(f"**Summary:** {paper['summary']}")
+        if paper['link']:
+            st.write(f"[Arxiv Link]({paper['link']})")
+        if paper['pdf']:
+            st.write(f"[PDF]({paper['pdf']})")
+
+        code_interpreter = st.checkbox(f"Code Interpreter for '{paper['title']}'", key=f"ci_{i}")
+        if code_interpreter:
+            code_task = st.text_area(
+                f"Describe the Python/Streamlit functionality to implement based on this paper:",
+                height=100, key=f"code_task_{i}"
+            )
+            if st.button(f"Generate Code for Paper {i+1}", key=f"gen_code_{i}"):
+                if code_task.strip():
+                    # Prompt the model to generate code
+                    code_prompt = f"""
+                    You are a coding assistant. 
+                    The user has a research paper titled: "{paper['title']}" 
+                    and summary: "{paper['summary']}".
+                    The user wants the following functionality implemented in Python with Streamlit and possible HTML5 components:
+                    "{code_task}"
+
+                    Requirements:
+                    - The code should be self-contained Python code, runnable within this Streamlit environment.
+                    - It should use `streamlit` library for UI and `print()` for textual outputs.
+                    - Provide only the Python code block, do not include extra explanations.
+                    """
+
+                    completion = openai_client.chat.completions.create(
+                        model=st.session_state["openai_model"],
+                        messages=[
+                            {"role": "system", "content": "You are a helpful coding assistant."},
+                            {"role": "user", "content": code_prompt}
+                        ],
+                        temperature=0.0
+                    )
+                    generated_code = completion.choices[0].message.content
+
+                    st.write("### Generated Code")
+                    st.code(generated_code, language="python")
+
+                    # Execute the generated code
+                    exec_locals = {}
+                    original_stdout = sys.stdout
+                    redirected_output = io.StringIO()
+                    sys.stdout = redirected_output
+                    try:
+                        exec(generated_code, {}, exec_locals)
+                    except Exception as e:
+                        st.error(f"Error running generated code: {e}")
+                    finally:
+                        sys.stdout = original_stdout
+
+                    code_output = redirected_output.getvalue()
+                    st.write("### Code Output")
+                    st.write(code_output)
+
+                    # TTS on code output
+                    if code_output.strip():
+                        audio_file = speak_with_edge_tts(code_output)
+                        if audio_file:
+                            play_and_download_audio(audio_file)
+
     return result
 
 def process_with_gpt(text):
@@ -335,7 +430,8 @@ def process_with_gpt(text):
         )
         ans = c.choices[0].message.content
         st.write("GPT-4o: " + ans)
-        create_file(text, ans, "md")
+        filename = generate_filename(ans.strip() if ans.strip() else text.strip(), "md")
+        create_file(filename, text, ans, should_save=True)
         st.session_state.messages.append({"role":"assistant","content":ans})
     return ans
 
@@ -352,11 +448,12 @@ def process_with_claude(text):
         )
         ans = r.content[0].text
         st.write("Claude-3.5: " + ans)
-        create_file(text, ans, "md")
+        filename = generate_filename(ans.strip() if ans.strip() else text.strip(), "md")
+        create_file(filename, text, ans, should_save=True)
         st.session_state.chat_history.append({"user":text,"claude":ans})
     return ans
 
-# 📂 10. File Management
+# 10. File Management
 def create_zip_of_files(md_files, mp3_files):
     """Create zip with intelligent naming"""
     md_files = [f for f in md_files if os.path.basename(f).lower() != 'readme.md']
@@ -364,7 +461,6 @@ def create_zip_of_files(md_files, mp3_files):
     if not all_files:
         return None
 
-    # Collect content for high-info term extraction
     all_content = []
     for f in all_files:
         if f.endswith('.md'):
@@ -445,7 +541,10 @@ def display_file_manager_sidebar(groups, sorted_prefixes):
         if st.button("⬇️ Zip All"):
             z = create_zip_of_files(all_md, all_mp3)
             if z:
-                st.sidebar.markdown(get_download_link(z),unsafe_allow_html=True)
+                with open(z, "rb") as f:
+                    b64 = base64.b64encode(f.read()).decode()
+                dl_link = f'<a href="data:file/zip;base64,{b64}" download="{os.path.basename(z)}">📂 Download {os.path.basename(z)}</a>'
+                st.sidebar.markdown(dl_link,unsafe_allow_html=True)
 
     for prefix in sorted_prefixes:
         files = groups[prefix]
@@ -468,7 +567,62 @@ def display_file_manager_sidebar(groups, sorted_prefixes):
                 ctime = datetime.fromtimestamp(os.path.getmtime(f)).strftime("%Y-%m-%d %H:%M:%S")
                 st.write(f"**{fname}** - {ctime}")
 
-# 🎯 11. Main Application
+# 11. New create_file function
+context = {}  # Execution context for code blocks
+
+def create_file(filename, prompt, response, should_save=True):
+    if not should_save:
+        return
+
+    base_filename, ext = os.path.splitext(filename)
+    combined_content = ""
+
+    # Add Prompt with markdown title and emoji
+    combined_content += "# Prompt 📝\n" + prompt + "\n\n"
+
+    # Add Response with markdown title and emoji
+    combined_content += "# Response 💬\n" + response + "\n\n"
+
+    # Check for code blocks in the response
+    resources = re.findall(r"```([\s\S]*?)```", response)
+    for resource in resources:
+        # Check if the resource contains Python code
+        if "python" in resource.lower():
+            cleaned_code = re.sub(r'^\s*python', '', resource, flags=re.IGNORECASE | re.MULTILINE)
+
+            # Add Code Results title with markdown and emoji
+            combined_content += "# Code Results 🚀\n"
+
+            original_stdout = sys.stdout
+            sys.stdout = io.StringIO()
+
+            try:
+                exec(cleaned_code, context)
+                code_output = sys.stdout.getvalue()
+                combined_content += f"```\n{code_output}\n```\n\n"
+                realtimeEvalResponse = "# Code Results 🚀\n" + "```" + code_output + "```\n\n"
+                st.code(realtimeEvalResponse)
+            except Exception as e:
+                combined_content += f"```python\nError executing Python code: {e}\n```\n\n"
+
+            sys.stdout = original_stdout
+        else:
+            # Add non-Python resources with markdown and emoji
+            combined_content += "# Resource 🛠️\n" + "```" + resource + "```\n\n"
+
+    # Save the combined content to a Markdown file
+    if should_save:
+        with open(f"{base_filename}.md", 'w') as file:
+            file.write(combined_content)
+            st.code(combined_content)
+
+    # Create a Base64 encoded link for the file
+    with open(f"{base_filename}.md", 'rb') as file:
+        encoded_file = base64.b64encode(file.read()).decode()
+        href = f'<a href="data:file/markdown;base64,{encoded_file}" download="{filename}">Download File 📄</a>'
+        st.markdown(href, unsafe_allow_html=True)
+
+# 12. Main Application
 def main():
     st.sidebar.markdown("### 🚲BikeAI🏆 Multi-Agent Research AI")
     tab_main = st.radio("Action:",["🎤 Voice Input","📸 Media Gallery","🔍 Search ArXiv","📝 File Editor"],horizontal=True)
@@ -600,7 +754,10 @@ def main():
             elif ext == "mp3":
                 st.audio(f)
             else:
-                st.markdown(get_download_link(f), unsafe_allow_html=True)
+                with open(f, "rb") as file:
+                    b64 = base64.b64encode(file.read()).decode()
+                dl_link = f'<a href="data:file/{ext};base64,{b64}" download="{fname}">Download {fname}</a>'
+                st.markdown(dl_link, unsafe_allow_html=True)
         if st.button("Close Group View"):
             st.session_state.viewing_prefix = None
 
@@ -609,4 +766,4 @@ def main():
         st.rerun()
 
 if __name__=="__main__":
-    main()
+    main()