Update app.py

app.py

@@ -7,35 +7,33 @@ import base64
 import pandas as pd
 import zipfile
 import PyPDF2
-import os  # Wird für die zukünftige Umgebungsvariablen-Nutzung bereitgehalten
 
-# --- Konfiguration ---
-st.set_page_config(page_title="OpenRouter Free Interface", layout="wide")
+# --- Page Config ---
+st.set_page_config(page_title="OpenRouter + Gemini AI Chat", layout="wide", initial_sidebar_state="expanded")
+
 OPENROUTER_API_BASE = "https://openrouter.ai/api/v1"
 
-# --- Page Title ---
-st.title("💸 OpenRouter Free-Tier Interface")
+# --- Title ---
+st.title("🤖 OpenRouter + Gemini AI Chat Interface")
 st.markdown("""
-**Willkommen im All-OpenRouter-Free-Interface Deluxe!**
-Chatte mit **kostenlosen (Free-Tier)** Modellen über die OpenRouter API.
-Alle Modelle unterliegen den OpenRouter-Ratenbegrenzungen.
+**Chat with Free-Tier OpenRouter models and optionally upload files (Text, PDF, ZIP, Images) for context.**
+💡 Responses are copyable with a single click.
 """)
 
-# --- Session State Management ---
+# --- Session State ---
 if "messages" not in st.session_state:
     st.session_state.messages = []
 if "uploaded_content" not in st.session_state:
     st.session_state.uploaded_content = None
 
-
-# --- Datei-Verarbeitung ---
+# --- Utilities ---
 def encode_image(image):
     buf = io.BytesIO()
     image.save(buf, format="JPEG")
     return base64.b64encode(buf.getvalue()).decode("utf-8")
 
-
 def process_file(uploaded_file):
+    """Process uploaded file (text, image, PDF, ZIP)"""
     file_type = uploaded_file.name.split('.')[-1].lower()
     text_exts = ('.txt', '.csv', '.py', '.html', '.js', '.css', '.json', '.xml', '.sql', '.xlsx')
 
@@ -50,14 +48,14 @@ def process_file(uploaded_file):
             df = pd.read_csv(uploaded_file) if file_type == "csv" else pd.read_excel(uploaded_file)
             return {"type": "text", "content": df.to_string()}
         except Exception as e:
-            return {"type": "error", "content": f"Fehler beim Lesen der Tabelle: {e}"}
+            return {"type": "error", "content": f"Failed reading table: {e}"}
 
     if file_type == "pdf":
         try:
             reader = PyPDF2.PdfReader(uploaded_file)
             return {"type": "text", "content": "".join(page.extract_text() or "" for page in reader.pages)}
         except Exception as e:
-            return {"type": "error", "content": f"PDF Fehler: {e}"}
+            return {"type": "error", "content": f"PDF Error: {e}"}
 
     if file_type == "zip":
         try:
@@ -67,16 +65,14 @@ def process_file(uploaded_file):
                     if not f.is_dir() and f.filename.lower().endswith(text_exts):
                         content += f"\n📄 {f.filename}:\n"
                         content += z.read(f.filename).decode("utf-8", errors="ignore")
-                return {"type": "text", "content": content or "ZIP enthält keine lesbaren Textdateien."}
+                return {"type": "text", "content": content or "ZIP contains no readable text files."}
         except Exception as e:
-            return {"type": "error", "content": f"ZIP Fehler: {e}"}
+            return {"type": "error", "content": f"ZIP Error: {e}"}
 
-    return {"type": "error", "content": "Nicht unterstütztes Dateiformat."}
+    return {"type": "error", "content": "Unsupported file type."}
 
-
-# --- Context-Length Fetch ---
 def fetch_model_contexts(api_key):
-    """Lädt alle Modelle + deren context_length."""
+    """Fetch context lengths for models"""
     if not api_key:
         return {}
     headers = {"Authorization": f"Bearer {api_key}"}
@@ -85,21 +81,45 @@ def fetch_model_contexts(api_key):
         contexts = {}
         if res.status_code == 200:
             for m in res.json().get("data", []):
-                mid = m.get("id")
-                ctx = m.get("context_length", 4096)
-                contexts[mid] = ctx
+                contexts[m.get("id")] = m.get("context_length", 4096)
         return contexts
     except Exception as e:
-        st.warning(f"⚠️ Fehler beim Laden der Modellinfos: {e}")
+        st.warning(f"⚠️ Failed to fetch model info: {e}")
         return {}
 
+def call_openrouter(model, messages, temp, max_tok, key):
+    headers = {
+        "Authorization": f"Bearer {key}",
+        "Content-Type": "application/json",
+        "Referer": "https://aicodecraft.io",
+        "X-Title": "OpenRouter-Free-Interface",
+    }
+    payload = {
+        "model": model,
+        "messages": messages,
+        "temperature": temp,
+        "max_tokens": max_tok,
+    }
+
+    res = requests.post(f"{OPENROUTER_API_BASE}/chat/completions", headers=headers, data=json.dumps(payload))
+    if res.status_code == 200:
+        try:
+            return res.json()["choices"][0]["message"]["content"]
+        except (KeyError, IndexError):
+            raise Exception("Invalid API response")
+    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}")
 
 # --- Sidebar ---
 with st.sidebar:
     st.header("⚙️ API Settings")
     api_key = st.text_input("OpenRouter API Key", type="password")
 
-    # Free Modelle (Fallback)
     FREE_MODEL_LIST = [
         "cognitivecomputations/dolphin-mistral-24b-venice-edition:free",
         "deepseek/deepseek-chat-v3",
@@ -109,32 +129,23 @@ with st.sidebar:
         "nousresearch/nous-hermes-2-mixtral-8x7b-dpo",
     ]
 
-    model = st.selectbox("Wähle ein Modell", FREE_MODEL_LIST, index=0)
+    model = st.selectbox("Select a model", FREE_MODEL_LIST, index=0)
 
-    # Context automatisch anpassen
+    # Context-length slider
     model_contexts = fetch_model_contexts(api_key)
     default_ctx = model_contexts.get(model, 4096)
+    max_tokens = st.slider(f"Max Tokens (max {default_ctx})", 1, min(default_ctx, 32000), min(512, default_ctx))
 
     temperature = st.slider("Temperature", 0.0, 1.0, 0.7)
-    max_tokens = st.slider(f"Max Tokens (max {default_ctx})", 1, min(default_ctx, 32000), min(512, default_ctx))
-    st.caption(f"🔢 Model Context Length: {default_ctx}")
 
-    if st.button("🔄 Chat Reset"):
+    if st.button("🔄 Reset Chat"):
         st.session_state.messages = []
         st.session_state.uploaded_content = None
-        st.success("Chat-Verlauf und Anhang gelöscht.")
-
-    st.markdown("""
-    ---
-    🧠 **Hinweis:** Diese Modelle sind **kostenlos**, aber ggf. durch Rate-Limits beschränkt.  
-    Dein API-Key wird nur **lokal** verwendet.
-    """)
-
-
-# --- Datei Upload ---
-uploaded_file = st.file_uploader("Upload File (optional)",
-                                 type=["jpg", "jpeg", "png", "txt", "pdf", "zip", "csv", "xlsx", "html", "css", "js", "py"])
+        st.success("Chat and attachment cleared!")
 
+# --- File Upload ---
+uploaded_file = st.file_uploader("Upload File (optional)", 
+                                 type=["jpg","jpeg","png","txt","pdf","zip","csv","xlsx","html","css","js","py"])
 if uploaded_file and st.session_state.uploaded_content is None:
     st.session_state.uploaded_content = process_file(uploaded_file)
 
@@ -142,7 +153,7 @@ if st.session_state.uploaded_content:
     processed = st.session_state.uploaded_content
     st.subheader("📎 Current Attachment:")
     if processed["type"] == "image":
-        st.image(processed["content"], caption="Attached Image", width=300)
+        st.image(processed["content"], width=300)
     elif processed["type"] == "text":
         st.text_area("File Preview", processed["content"], height=150)
     elif processed["type"] == "error":
@@ -151,47 +162,15 @@ if st.session_state.uploaded_content:
         st.session_state.uploaded_content = None
         st.experimental_rerun()
 
-
-# --- Chat Verlauf anzeigen ---
+# --- Chat History ---
 for msg in st.session_state.messages:
     with st.chat_message(msg["role"]):
         st.markdown(msg["content"])
 
-
-# --- API Call ---
-def call_openrouter(model, messages, temp, max_tok, key):
-    headers = {
-        "Authorization": f"Bearer {key}",
-        "Content-Type": "application/json",
-        "Referer": "https://aicodecraft.io",
-        "X-Title": "OpenRouter-Free-Interface",
-    }
-    payload = {
-        "model": model,
-        "messages": messages,
-        "temperature": temp,
-        "max_tokens": max_tok,
-    }
-
-    res = requests.post(f"{OPENROUTER_API_BASE}/chat/completions", headers=headers, data=json.dumps(payload))
-    if res.status_code == 200:
-        try:
-            return res.json()["choices"][0]["message"]["content"]
-        except (KeyError, IndexError):
-            raise Exception("Fehlerhafte API-Antwort: Konnte Antworttext nicht extrahieren.")
-    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}")
-
-
 # --- Chat Input ---
-if prompt := st.chat_input("Deine Nachricht..."):
+if prompt := st.chat_input("Your message..."):
     if not api_key:
-        st.warning("Bitte trage deinen OpenRouter API Key in der Sidebar ein.")
+        st.warning("Please enter your OpenRouter API Key in the sidebar.")
         st.stop()
 
     st.session_state.messages.append({"role": "user", "content": prompt})
@@ -200,6 +179,7 @@ if prompt := st.chat_input("Deine Nachricht..."):
 
     messages = [{"role": m["role"], "content": m["content"]} for m in st.session_state.messages]
 
+    # Attach file if exists
     if st.session_state.uploaded_content:
         content = st.session_state.uploaded_content
         if content["type"] == "image":
@@ -211,11 +191,21 @@ if prompt := st.chat_input("Deine Nachricht..."):
         elif content["type"] == "text":
             messages[-1]["content"] += f"\n\n[Attached File Content]\n{content['content']}"
 
+    # Generate response
     with st.chat_message("assistant"):
-        with st.spinner(f"Fragend {model}..."):
+        with st.spinner(f"Asking {model}..."):
             try:
                 reply = call_openrouter(model, messages, temperature, max_tokens, api_key)
-                st.markdown(reply)
+                
+                # Clipboard-ready response with JS
+                st.markdown(f"""
+                <div style="position: relative;">
+                    <button onclick="navigator.clipboard.writeText(`{reply.replace('`','\\`')}`)"
+                            style="position:absolute; right:0; top:0;">📋 Copy</button>
+                    <div style="padding-right:50px;">{reply}</div>
+                </div>
+                """, unsafe_allow_html=True)
+                
                 st.session_state.messages.append({"role": "assistant", "content": reply})
             except Exception as e:
                 st.error(str(e))