ai_text

Running

App Files Files Community

Pepguy commited on Oct 6, 2025

Commit

447ef90

verified ·

1 Parent(s): 03bf406

Update app.py

Browse files

Files changed (1) hide show

app.py +85 -103

app.py CHANGED Viewed

@@ -2,9 +2,8 @@
 # pip install flask google-genai requests boto3
 import os
-import io
 import time
-import base64
 import requests
 from flask import Flask, request, render_template_string, jsonify
 from google import genai
@@ -12,161 +11,144 @@ from google.genai import types
 app = Flask(__name__)
-# Lambda endpoint for history (your Lambda function URL)
 LAMBDA_URL = os.getenv("LAMBDA_URL", "https://your-lambda-function-url")
-# Google Gemini API setup
-client = genai.Client(api_key=os.getenv("GEMINI_API_KEY", ""))
 HTML = """
 <!DOCTYPE html>
 <html lang="en">
-<head>
-  <meta charset="UTF-8" />
-  <title>Gemini 2.0 Flash Test</title>
-</head>
 <body style="font-family:sans-serif;padding:2rem;max-width:700px;">
-  <h1>Gemini 2.0 Flash (Text + Image)</h1>
-  <label><input type="checkbox" id="historyToggle"/> Enable History</label>
-  <br/><br/>
   <form id="genai-form">
-    <textarea id="prompt" rows="5" cols="60" placeholder="Enter your prompt here"></textarea><br/><br/>
     <input type="file" id="imageInput" accept="image/*"/><br/><br/>
     <button type="submit">Generate</button>
   </form>
   <pre id="output" style="background:#f4f4f4;padding:1rem;margin-top:1rem;white-space:pre-wrap;"></pre>
   <script>
   const form = document.getElementById('genai-form');
-  const toggle = document.getElementById('historyToggle');
   const out = document.getElementById('output');
   form.addEventListener('submit', async e => {
     e.preventDefault();
     const prompt = document.getElementById('prompt').value.trim();
     const fileInput = document.getElementById('imageInput');
-    const historyOn = toggle.checked;
-    if (!prompt && fileInput.files.length === 0) {
-      out.textContent = 'Please enter a prompt or attach an image.';
-      return;
-    }
-    out.textContent = 'Generating…';
     const formData = new FormData();
     formData.append("text", prompt);
-    formData.append("history", historyOn);
-    if (fileInput.files.length > 0) {
-      formData.append("image", fileInput.files[0]);
-    }
     try {
-      const resp = await fetch('/generate', {
-        method: 'POST',
-        body: formData
-      });
       const data = await resp.json();
-      if (data.error) {
-        out.textContent = 'Error: ' + data.error;
-      } else {
-        out.textContent =
-          "🕒 Total Time: " + data.timing.total_ms + " ms\\n" +
-          "⚙️ Model Time: " + data.timing.model_ms + " ms\\n\\n" +
-          "📜 Result:\\n" + data.result +
-          (data.history ? "\\n\\n💾 History synced" : "");
-      }
     } catch (err) {
-      console.error(err);
       out.textContent = 'Fetch error: ' + err.message;
     }
   });
   </script>
-</body>
-</html>
 """
-def generate_from_gemini(prompt: str, image_bytes: bytes = None):
-    """Send text + optional image to Gemini and return response + timing."""
     start_time = time.time()
     parts = []
-    if prompt:
-        parts.append(types.Part.from_text(text=prompt))
-    if image_bytes:
-        parts.append(types.Part.from_bytes(data=image_bytes, mime_type="image/jpeg"))
     contents = [types.Content(role="user", parts=parts)]
-    config = types.GenerateContentConfig(response_mime_type="text/plain")
     model_start = time.time()
-    response = client.models.generate_content(
-        model="gemini-2.5-flash-lite",
-        #model="gemini-2.0-flash",
-        contents=contents,
-        config=config,
-    )
     model_end = time.time()
-    return {
-        "text": response.text,
-        "timing": {
-            "total_ms": int((time.time() - start_time) * 1000),
-            "model_ms": int((model_end - model_start) * 1000),
-        },
-    }
-def handle_history_sync(history_data):
-    """Send and retrieve chat history from Lambda (DynamoDB backend)."""
-    try:
-        resp = requests.post(LAMBDA_URL, json=history_data, timeout=5)
-        resp.raise_for_status()
-        return resp.json()
-    except Exception as e:
-        app.logger.warning(f"History sync failed: {e}")
-        return None
-@app.route('/')
 def index():
     return render_template_string(HTML)
-@app.route('/generate', methods=['POST'])
 def gen():
-    prompt = request.form.get("text", "")
-    history_on = request.form.get("history", "false") == "true"
     image = request.files.get("image")
-    image_bytes = image.read() if image else None
-    if not prompt and not image_bytes:
         return jsonify({"error": "No prompt or image provided"}), 400
     try:
-        result = generate_from_gemini(prompt, image_bytes)
-        response_json = {
-            "result": result["text"],
-            "timing": result["timing"],
-        }
-        if history_on:
-            history_payload = {
-                "prompt": prompt,
-                "response": result["text"],
-                "timestamp": time.time(),
-            }
-            history_resp = handle_history_sync(history_payload)
-            if history_resp:
-                response_json["history"] = history_resp
-        return jsonify(response_json)
     except Exception as e:
         app.logger.exception("Generation failed")
         return jsonify({"error": str(e)}), 500
 if __name__ == "__main__":
-    port = int(os.environ.get("PORT", 7860))
     app.run(host="0.0.0.0", port=port)

 # pip install flask google-genai requests boto3
 import os
 import time
+import threading
 import requests
 from flask import Flask, request, render_template_string, jsonify
 from google import genai
 app = Flask(__name__)
+# --- Configuration ---
 LAMBDA_URL = os.getenv("LAMBDA_URL", "https://your-lambda-function-url")
+GEMINI_KEY = os.getenv("GEMINI_API_KEY", "")
+FLUSH_INTERVAL = 30  # seconds between DB backups per user
+client = genai.Client(api_key=GEMINI_KEY)
+user_memory = {}  # { user_id: { "history": [], "last_sync": timestamp } }
+# --- Background thread for periodic flush ---
+def flush_loop():
+    while True:
+        now = time.time()
+        for uid, data in list(user_memory.items()):
+            if now - data.get("last_sync", 0) >= FLUSH_INTERVAL and data["history"]:
+                try:
+                    payload = {"user_id": uid, "history": data["history"]}
+                    requests.post(LAMBDA_URL, json=payload, timeout=5)
+                    user_memory[uid]["last_sync"] = now
+                    app.logger.info(f"Synced memory for {uid}")
+                except Exception as e:
+                    app.logger.warning(f"Failed sync for {uid}: {e}")
+        time.sleep(5)
+threading.Thread(target=flush_loop, daemon=True).start()
+# --- HTML Frontend ---
 HTML = """
 <!DOCTYPE html>
 <html lang="en">
+<head><meta charset="UTF-8" /><title>Gemini 2.5 Flash-Lite</title></head>
 <body style="font-family:sans-serif;padding:2rem;max-width:700px;">
+  <h1>Gemini 2.5 Flash-Lite (Text + Image)</h1>
   <form id="genai-form">
+    <input type="text" id="userId" placeholder="User ID / Token" style="width:300px;" /><br/><br/>
+    <textarea id="prompt" rows="5" cols="60" placeholder="Enter prompt..."></textarea><br/><br/>
     <input type="file" id="imageInput" accept="image/*"/><br/><br/>
     <button type="submit">Generate</button>
   </form>
   <pre id="output" style="background:#f4f4f4;padding:1rem;margin-top:1rem;white-space:pre-wrap;"></pre>
   <script>
   const form = document.getElementById('genai-form');
   const out = document.getElementById('output');
   form.addEventListener('submit', async e => {
     e.preventDefault();
     const prompt = document.getElementById('prompt').value.trim();
+    const uid = document.getElementById('userId').value.trim();
     const fileInput = document.getElementById('imageInput');
+    if (!uid) { out.textContent = 'Please enter a user ID/token.'; return; }
+    if (!prompt && fileInput.files.length === 0) { out.textContent = 'Enter text or attach image.'; return; }
+    out.textContent = 'Generating…';
     const formData = new FormData();
     formData.append("text", prompt);
+    formData.append("user_id", uid);
+    if (fileInput.files.length > 0) formData.append("image", fileInput.files[0]);
     try {
+      const resp = await fetch('/generate', { method: 'POST', body: formData });
       const data = await resp.json();
+      if (data.error) out.textContent = 'Error: ' + data.error;
+      else out.textContent =
+        "🕒 Total Time: " + data.timing.total_ms + " ms\\n" +
+        "⚙️ Model Time: " + data.timing.model_ms + " ms\\n\\n" +
+        "📜 Result:\\n" + data.result;
     } catch (err) {
       out.textContent = 'Fetch error: ' + err.message;
     }
   });
   </script>
+</body></html>
 """
+# --- Gemini Generation ---
+def generate_from_gemini(prompt, image_bytes=None):
     start_time = time.time()
     parts = []
+    if prompt: parts.append(types.Part.from_text(text=prompt))
+    if image_bytes: parts.append(types.Part.from_bytes(data=image_bytes, mime_type="image/jpeg"))
     contents = [types.Content(role="user", parts=parts)]
+    cfg = types.GenerateContentConfig(response_mime_type="text/plain")
     model_start = time.time()
+    res = client.models.generate_content(model="gemini-2.5-flash-lite", contents=contents, config=cfg)
     model_end = time.time()
+    return {"text": res.text, "timing": {
+        "total_ms": int((time.time() - start_time)*1000),
+        "model_ms": int((model_end - model_start)*1000)
+    }}
+# --- History Management ---
+def get_user_history(uid):
+    if uid not in user_memory:
+        try:
+            resp = requests.get(f"{LAMBDA_URL}?user_id={uid}", timeout=5)
+            resp.raise_for_status()
+            user_memory[uid] = {"history": resp.json().get("history", []), "last_sync": 0}
+            app.logger.info(f"Loaded history for {uid}")
+        except Exception as e:
+            app.logger.warning(f"Failed to load history for {uid}: {e}")
+            user_memory[uid] = {"history": [], "last_sync": 0}
+    return user_memory[uid]["history"]
+def update_user_history(uid, prompt, response):
+    entry = {"prompt": prompt, "response": response, "timestamp": time.time()}
+    user_memory.setdefault(uid, {"history": [], "last_sync": 0})["history"].append(entry)
+# --- Routes ---
+@app.route("/")
 def index():
     return render_template_string(HTML)
+@app.route("/generate", methods=["POST"])
 def gen():
+    uid = request.form.get("user_id", "").strip()
+    if not uid:
+        return jsonify({"error": "Missing user ID/token"}), 400
+    prompt = request.form.get("text", "")
     image = request.files.get("image")
+    img_bytes = image.read() if image else None
+    if not prompt and not img_bytes:
         return jsonify({"error": "No prompt or image provided"}), 400
     try:
+        _ = get_user_history(uid)
+        result = generate_from_gemini(prompt, img_bytes)
+        update_user_history(uid, prompt, result["text"])
+        return jsonify({"result": result["text"], "timing": result["timing"]})
     except Exception as e:
         app.logger.exception("Generation failed")
         return jsonify({"error": str(e)}), 500
 if __name__ == "__main__":
+    port = int(os.getenv("PORT", 7860))
     app.run(host="0.0.0.0", port=port)