Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,433 @@
-import altair as alt
-import numpy as np
-import pandas as pd
+import os
 import streamlit as st
+import tempfile
+import requests
+import json
+from datetime import datetime
 
-"""
-# Welcome to Streamlit!
+# ----------------------------------------------------
+# 🚨 HUGGINGFACE SPACES FIX
+# ----------------------------------------------------
+TEMP_STREAMLIT_HOME = os.path.join(tempfile.gettempdir(), "st_config_workaround")
+os.makedirs(TEMP_STREAMLIT_HOME, exist_ok=True)
 
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
+os.environ["STREAMLIT_HOME"] = TEMP_STREAMLIT_HOME
+os.environ["STREAMLIT_GATHER_USAGE_STATS"] = "false"
 
-In the meantime, below is an example of what you can do with just a few lines of code:
+CONFIG_PATH = os.path.join(TEMP_STREAMLIT_HOME, "config.toml")
+CONFIG_CONTENT = """
+[browser]
+gatherUsageStats = false
 """
 
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+if not os.path.exists(CONFIG_PATH):
+    try:
+        with open(CONFIG_PATH, "w") as f:
+            f.write(CONFIG_CONTENT)
+    except Exception:
+        pass
+
+# ----------------------------------------------------
+# CONFIG
+# ----------------------------------------------------
+st.set_page_config(
+    page_title="AI Code Forge", 
+    layout="wide",
+    page_icon="⚒️"
+)
+
+OPENROUTER_API_BASE = "https://openrouter.ai/api/v1"
+
+FREE_MODELS = {
+    "🧠 DeepSeek Chat": "deepseek/deepseek-chat-v3.1:free",
+    "💻 Qwen Coder": "qwen/qwen3-coder:free",
+    "🔮 Gemma 27B": "google/gemma-3-27b-it:free",
+    "🐬 Dolphin Mistral": "cognitivecomputations/dolphin-mistral-24b-venice-edition:free",
+    "⚡ Nemotron Nano": "nvidia/nemotron-nano-9b-v2:free",
+}
+
+# ----------------------------------------------------
+# SESSION STATE
+# ----------------------------------------------------
+if "messages" not in st.session_state:
+    st.session_state.messages = {}
+if "current_tool" not in st.session_state:
+    st.session_state.current_tool = "home"
+if "last_output" not in st.session_state:
+    st.session_state.last_output = ""
+if "dark_mode" not in st.session_state:
+    st.session_state.dark_mode = True
+
+# ----------------------------------------------------
+# HELPER FUNCTIONS
+# ----------------------------------------------------
+def fetch_model_contexts(api_key):
+    if not api_key:
+        return {}
+    headers = {"Authorization": f"Bearer {api_key}"}
+    try:
+        res = requests.get(f"{OPENROUTER_API_BASE}/models", headers=headers, timeout=10)
+        contexts = {}
+        if res.status_code == 200:
+            for m in res.json().get("data", []):
+                contexts[m.get("id")] = m.get("context_length", 4096)
+        return contexts
+    except Exception:
+        return {}
+
+def call_openrouter(model, messages, temp, max_tok, key, system_prompt=None):
+    headers = {
+        "Authorization": f"Bearer {key}",
+        "Content-Type": "application/json",
+        "Referer": "https://aicodecraft.io",
+        "X-Title": "AI-Code-Forge",
+    }
+    
+    api_messages = messages.copy()
+    if system_prompt:
+        api_messages.insert(0, {"role": "system", "content": system_prompt})
+    
+    payload = {
+        "model": model,
+        "messages": api_messages,
+        "temperature": temp,
+        "max_tokens": max_tok,
+    }
+    
+    res = requests.post(
+        f"{OPENROUTER_API_BASE}/chat/completions", 
+        headers=headers, 
+        data=json.dumps(payload),
+        timeout=60
+    )
+    
+    if res.status_code == 200:
+        try:
+            return res.json()["choices"][0]["message"]["content"]
+        except (KeyError, IndexError):
+            raise Exception("Fehlerhafte API-Antwort")
+    else:
+        try:
+            err = res.json()
+            msg = err.get("error", {}).get("message", res.text)
+        except:
+            msg = res.text
+        raise Exception(f"API Error {res.status_code}: {msg}")
+
+def extract_code_blocks(text):
+    """Extrahiert Code aus Markdown Code-Blöcken"""
+    import re
+    pattern = r"```(\w+)?\n(.*?)```"
+    matches = re.findall(pattern, text, re.DOTALL)
+    if matches:
+        return [(lang or "text", code.strip()) for lang, code in matches]
+    return [("text", text)]
+
+# ----------------------------------------------------
+# TOOL PRESETS
+# ----------------------------------------------------
+TOOL_PRESETS = {
+    "code_gen": {
+        "icon": "🎨",
+        "title": "Code Generator",
+        "color": "#00ff88",
+        "system_prompt": "Du bist ein Expert Code Generator. Erstelle sauberen, produktionsreifen Code mit Kommentaren. Antworte NUR mit Code in Markdown Code-Blöcken.",
+        "templates": {
+            "Flask REST API": "Erstelle eine Flask REST API mit:\n- CRUD Endpoints für 'tasks'\n- SQLAlchemy Models\n- Error Handling\n- CORS aktiviert",
+            "React Component": "Erstelle eine React Functional Component mit:\n- Props für title und items (Array)\n- useState für selected item\n- Tailwind CSS Styling\n- Click Handler",
+            "Python CLI Tool": "Erstelle ein Python CLI Tool mit:\n- argparse für Arguments\n- Logging\n- Error Handling\n- Main Guard",
+            "FastAPI Endpoint": "Erstelle FastAPI Endpoints:\n- POST /users - Create User\n- GET /users/{id} - Get User\n- Pydantic Models\n- SQLAlchemy Integration",
+        }
+    },
+    "refactor": {
+        "icon": "🔄",
+        "title": "Code Refactorer",
+        "color": "#00aaff",
+        "system_prompt": "Du bist ein Code Refactoring Expert. Analysiere Code und verbessere: Lesbarkeit, Performance, Best Practices. Erkläre deine Änderungen kurz.",
+        "templates": {
+            "Clean Code": "Refactore diesen Code nach Clean Code Prinzipien:\n- Verbessere Naming\n- Reduziere Komplexität\n- Extrahiere Funktionen\n\n[Füge Code hier ein]",
+            "Performance": "Optimiere diesen Code für Performance:\n- Identifiziere Bottlenecks\n- Verbessere Algorithmen\n- Reduziere Memory Usage\n\n[Füge Code hier ein]",
+            "Type Safety": "Füge Type Hints/TypeScript hinzu:\n- Alle Funktionen\n- Variables wo sinnvoll\n- Return Types\n\n[Füge Code hier ein]",
+        }
+    },
+    "debug": {
+        "icon": "🐛",
+        "title": "Debug Helper",
+        "color": "#ff4444",
+        "system_prompt": "Du bist ein Debugging Expert. Analysiere Code/Errors systematisch. Erkläre das Problem und liefere Fix.",
+        "templates": {
+            "Error Analysis": "Ich habe diesen Error:\n[Error Message]\n\nIn diesem Code:\n[Code]\n\nWas ist das Problem und wie fixe ich es?",
+            "Logic Bug": "Dieser Code macht nicht was er soll:\n[Code]\n\nErwartetes Verhalten: [beschreiben]\nAktuelles Verhalten: [beschreiben]\n\nWo ist der Logic Bug?",
+            "Performance Issue": "Dieser Code ist zu langsam:\n[Code]\n\nInput Size: [beschreiben]\nAktuelle Runtime: [beschreiben]\n\nWo ist das Problem?",
+        }
+    },
+    "docs": {
+        "icon": "📚",
+        "title": "Doc Generator",
+        "color": "#ffaa00",
+        "system_prompt": "Du bist ein Technical Writer. Erstelle klare, strukturierte Dokumentation mit Beispielen.",
+        "templates": {
+            "Function Docs": "Erstelle Docstrings für diese Funktionen:\n[Code]\n\nFormat: Google Style für Python / JSDoc für JavaScript",
+            "README": "Erstelle ein README.md für dieses Projekt:\n- Beschreibung\n- Installation\n- Usage Examples\n- API Reference\n\nProjekt: [beschreiben]",
+            "API Docs": "Erstelle API Dokumentation für diese Endpoints:\n[Code/Routes]\n\nInkl. Request/Response Beispiele",
+        }
+    },
+    "explain": {
+        "icon": "💡",
+        "title": "Code Explainer",
+        "color": "#aa88ff",
+        "system_prompt": "Du bist ein Code Educator. Erkläre Code Schritt-für-Schritt, klar und verständlich.",
+        "templates": {
+            "ELI5": "Erkläre diesen Code als wäre ich 5:\n[Code]",
+            "Deep Dive": "Erkläre diesen Code im Detail:\n- Was macht er?\n- Wie funktioniert er?\n- Warum wurde es so gelöst?\n- Alternativen?\n\n[Code]",
+            "Architecture": "Erkläre die Architektur dieses Codes:\n- Design Patterns\n- Komponenten\n- Datenfluss\n\n[Code]",
+        }
+    }
+}
+
+# ----------------------------------------------------
+# SIDEBAR
+# ----------------------------------------------------
+with st.sidebar:
+    st.markdown("# ⚒️ AI Code Forge")
+    st.markdown("---")
+    
+    # API Settings
+    with st.expander("🔑 API Settings", expanded=True):
+        api_key = st.text_input("OpenRouter API Key", type="password")
+        
+        model_name = st.selectbox("Model", list(FREE_MODELS.keys()))
+        model = FREE_MODELS[model_name]
+        
+        model_contexts = fetch_model_contexts(api_key)
+        default_ctx = model_contexts.get(model, 4096)
+        
+        temperature = st.slider("Temperature", 0.0, 1.0, 0.7, 0.1)
+        max_tokens = st.slider(
+            "Max Tokens", 
+            256, 
+            min(default_ctx, 16000), 
+            min(2048, default_ctx),
+            step=256
+        )
+        
+        st.caption(f"📊 Context: {default_ctx:,} tokens")
+    
+    st.markdown("---")
+    
+    # Navigation
+    st.markdown("### 🧭 Tools")
+    
+    if st.button("🏠 Home", use_container_width=True):
+        st.session_state.current_tool = "home"
+        st.rerun()
+    
+    for tool_id, tool in TOOL_PRESETS.items():
+        if st.button(
+            f"{tool['icon']} {tool['title']}", 
+            use_container_width=True,
+            type="primary" if st.session_state.current_tool == tool_id else "secondary"
+        ):
+            st.session_state.current_tool = tool_id
+            if tool_id not in st.session_state.messages:
+                st.session_state.messages[tool_id] = []
+            st.rerun()
+    
+    st.markdown("---")
+    
+    # Actions
+    if st.session_state.current_tool != "home":
+        if st.button("🗑️ Clear Chat", use_container_width=True):
+            if st.session_state.current_tool in st.session_state.messages:
+                st.session_state.messages[st.session_state.current_tool] = []
+            st.success("Chat cleared!")
+            st.rerun()
+    
+    st.markdown("---")
+    st.caption("💸 Using Free Models")
+    st.caption(f"⏰ {datetime.now().strftime('%H:%M')}")
+
+# ----------------------------------------------------
+# MAIN CONTENT
+# ----------------------------------------------------
+
+# HOME
+if st.session_state.current_tool == "home":
+    st.markdown("# ⚒️ Welcome to AI Code Forge")
+    st.markdown("### Your AI-Powered Development Assistant")
+    
+    st.markdown("---")
+    
+    # Tool Cards
+    cols = st.columns(3)
+    tool_items = list(TOOL_PRESETS.items())
+    
+    for idx, (tool_id, tool) in enumerate(tool_items):
+        with cols[idx % 3]:
+            with st.container():
+                st.markdown(f"""
+                <div style="
+                    padding: 20px;
+                    border-radius: 10px;
+                    border-left: 4px solid {tool['color']};
+                    background: rgba(255,255,255,0.05);
+                    margin-bottom: 20px;
+                ">
+                    <h2>{tool['icon']} {tool['title']}</h2>
+                    <p style="opacity: 0.8;">{tool['system_prompt'][:100]}...</p>
+                </div>
+                """, unsafe_allow_html=True)
+                
+                if st.button(f"Open {tool['title']}", key=f"open_{tool_id}", use_container_width=True):
+                    st.session_state.current_tool = tool_id
+                    if tool_id not in st.session_state.messages:
+                        st.session_state.messages[tool_id] = []
+                    st.rerun()
+    
+    st.markdown("---")
+    
+    # Quick Stats
+    col1, col2, col3 = st.columns(3)
+    with col1:
+        st.metric("Free Models", len(FREE_MODELS))
+    with col2:
+        total_msgs = sum(len(msgs) for msgs in st.session_state.messages.values())
+        st.metric("Total Messages", total_msgs)
+    with col3:
+        st.metric("Tools", len(TOOL_PRESETS))
+
+# TOOL VIEW
+else:
+    tool = TOOL_PRESETS[st.session_state.current_tool]
+    
+    # Header
+    st.markdown(f"""
+    <div style="
+        padding: 20px;
+        border-radius: 10px;
+        border-left: 6px solid {tool['color']};
+        background: rgba(255,255,255,0.05);
+        margin-bottom: 30px;
+    ">
+        <h1>{tool['icon']} {tool['title']}</h1>
+        <p style="opacity: 0.8; font-size: 1.1em;">{tool['system_prompt']}</p>
+    </div>
+    """, unsafe_allow_html=True)
+    
+    # Templates
+    with st.expander("📋 Quick Templates", expanded=False):
+        for template_name, template_text in tool['templates'].items():
+            if st.button(f"📄 {template_name}", key=f"template_{template_name}"):
+                st.session_state[f"template_text_{st.session_state.current_tool}"] = template_text
+                st.rerun()
+    
+    # Chat Interface
+    st.markdown("### 💬 Chat")
+    
+    # Display messages
+    for msg in st.session_state.messages.get(st.session_state.current_tool, []):
+        with st.chat_message(msg["role"]):
+            if msg["role"] == "assistant":
+                # Check for code blocks
+                code_blocks = extract_code_blocks(msg["content"])
+                if len(code_blocks) == 1 and code_blocks[0][0] == "text":
+                    st.markdown(msg["content"])
+                else:
+                    # Has code blocks
+                    parts = msg["content"].split("```")
+                    for i, part in enumerate(parts):
+                        if i % 2 == 0:
+                            # Text part
+                            if part.strip():
+                                st.markdown(part)
+                        else:
+                            # Code part
+                            lines = part.split("\n", 1)
+                            lang = lines[0].strip() if lines else "text"
+                            code = lines[1] if len(lines) > 1 else part
+                            
+                            col1, col2 = st.columns([6, 1])
+                            with col1:
+                                st.code(code, language=lang)
+                            with col2:
+                                if st.button("📋", key=f"copy_{i}_{msg.get('timestamp', 0)}"):
+                                    st.toast("Code copied! (simulation)")
+            else:
+                st.markdown(msg["content"])
+    
+    # Input
+    default_text = st.session_state.get(f"template_text_{st.session_state.current_tool}", "")
+    if default_text:
+        del st.session_state[f"template_text_{st.session_state.current_tool}"]
+    
+    if prompt := st.chat_input("Your message...", key=f"input_{st.session_state.current_tool}"):
+        user_prompt = prompt
+    elif default_text:
+        user_prompt = default_text
+    else:
+        user_prompt = None
+    
+    if user_prompt:
+        if not api_key:
+            st.warning("⚠️ Please enter your OpenRouter API Key in the sidebar!")
+            st.stop()
+        
+        # Add user message
+        st.session_state.messages[st.session_state.current_tool].append({
+            "role": "user", 
+            "content": user_prompt,
+            "timestamp": datetime.now().timestamp()
+        })
+        
+        with st.chat_message("user"):
+            st.markdown(user_prompt)
+        
+        # Prepare messages for API
+        api_messages = [
+            {"role": m["role"], "content": m["content"]} 
+            for m in st.session_state.messages[st.session_state.current_tool]
+        ]
+        
+        # Generate response
+        with st.chat_message("assistant"):
+            with st.spinner(f"🤖 {model_name} thinking..."):
+                try:
+                    reply = call_openrouter(
+                        model, 
+                        api_messages, 
+                        temperature, 
+                        max_tokens, 
+                        api_key,
+                        system_prompt=tool['system_prompt']
+                    )
+                    
+                    # Display with code highlighting
+                    code_blocks = extract_code_blocks(reply)
+                    if len(code_blocks) == 1 and code_blocks[0][0] == "text":
+                        st.markdown(reply)
+                    else:
+                        parts = reply.split("```")
+                        for i, part in enumerate(parts):
+                            if i % 2 == 0:
+                                if part.strip():
+                                    st.markdown(part)
+                            else:
+                                lines = part.split("\n", 1)
+                                lang = lines[0].strip() if lines else "text"
+                                code = lines[1] if len(lines) > 1 else part
+                                st.code(code, language=lang)
+                    
+                    st.session_state.messages[st.session_state.current_tool].append({
+                        "role": "assistant", 
+                        "content": reply,
+                        "timestamp": datetime.now().timestamp()
+                    })
+                    
+                except Exception as e:
+                    error_msg = f"❌ Error: {str(e)}"
+                    st.error(error_msg)
+                    st.session_state.messages[st.session_state.current_tool].append({
+                        "role": "assistant", 
+                        "content": error_msg,
+                        "timestamp": datetime.now().timestamp()
+                    })