ai_text_plus_audio

Sleeping

App Files Files Community

Pepguy commited on Oct 2, 2025

Commit

bbcf9c6

verified ·

1 Parent(s): 5be82da

Update app.py

Browse files

Files changed (1) hide show

app.py +31 -25

app.py CHANGED Viewed

@@ -1,4 +1,4 @@
-# To run this code you need to install the following dependencies:
 # pip install flask google-genai
 import os
@@ -17,8 +17,9 @@ HTML = """
 </head>
 <body style="font-family:sans-serif;padding:2rem;">
   <h1>Gemini-2.0-Flash Test</h1>
-  <form id="genai-form">
-    <textarea id="prompt" rows="6" cols="60" placeholder="Enter your prompt here"></textarea><br/><br/>
     <button type="submit">Generate</button>
   </form>
   <pre id="output" style="background:#f4f4f4;padding:1rem;margin-top:1rem;"></pre>
@@ -27,18 +28,17 @@ HTML = """
   const form = document.getElementById('genai-form');
   form.addEventListener('submit', async e => {
     e.preventDefault();
-    const prompt = document.getElementById('prompt').value.trim();
     const out = document.getElementById('output');
     out.textContent = 'Generating…';
     try {
       const resp = await fetch('/generate', {
         method: 'POST',
-        headers: { 'Content-Type': 'application/json' },
-        body: JSON.stringify({ text: prompt }),
       });
-      // if you want to debug status codes:
       if (!resp.ok) {
         const errText = await resp.text();
         throw new Error(`Server returned ${resp.status}: ${errText}`);
@@ -46,7 +46,6 @@ HTML = """
       const data = await resp.json();
       if (data.error) {
-        // show the backend error message
         out.textContent = 'Error: ' + data.error;
       } else if (data.result) {
         out.textContent = data.result;
@@ -63,27 +62,28 @@ HTML = """
 </html>
 """
-def generate_from_gemini(prompt: str) -> str:
     client = genai.Client(
-        api_key="AIzaSyDolbPUZBPUPvQUu-RGktJmvnUpkcEKIYo",
     )
     model = "gemini-2.0-flash"
-    contents = [
-        types.Content(
-            role="user",
-            parts=[types.Part.from_text(text=prompt)],
-        )
-    ]
     config = types.GenerateContentConfig(response_mime_type="text/plain")
-    # Synchronous, non-streaming call:
     response = client.models.generate_content(
         model=model,
         contents=contents,
         config=config,
     )
-    # The full text is in response.text
     return response.text
 @app.route('/')
@@ -92,14 +92,20 @@ def index():
 @app.route('/generate', methods=['POST'])
 def gen():
-    data = request.get_json(silent=True) or {}
-    prompt = data.get("text", "")
-    app.logger.info(f"Received prompt: {prompt!r}")
-    if not prompt:
-        return jsonify({"error": "No prompt provided"}), 400
     try:
-        output = generate_from_gemini(prompt)
-        app.logger.info("Generation succeeded, length=%d", len(output))
         return jsonify({"result": output})
     except Exception as e:
         app.logger.exception("Generation failed")

+# To run this code you need to install:
 # pip install flask google-genai
 import os
 </head>
 <body style="font-family:sans-serif;padding:2rem;">
   <h1>Gemini-2.0-Flash Test</h1>
+  <form id="genai-form" enctype="multipart/form-data">
+    <textarea id="prompt" name="text" rows="6" cols="60" placeholder="Enter your prompt here"></textarea><br/><br/>
+    <input type="file" id="image" name="image" accept="image/*" /><br/><br/>
     <button type="submit">Generate</button>
   </form>
   <pre id="output" style="background:#f4f4f4;padding:1rem;margin-top:1rem;"></pre>
   const form = document.getElementById('genai-form');
   form.addEventListener('submit', async e => {
     e.preventDefault();
     const out = document.getElementById('output');
     out.textContent = 'Generating…';
+    const formData = new FormData(form);
     try {
       const resp = await fetch('/generate', {
         method: 'POST',
+        body: formData
       });
       if (!resp.ok) {
         const errText = await resp.text();
         throw new Error(`Server returned ${resp.status}: ${errText}`);
       const data = await resp.json();
       if (data.error) {
         out.textContent = 'Error: ' + data.error;
       } else if (data.result) {
         out.textContent = data.result;
 </html>
 """
+def generate_from_gemini(prompt: str, image_bytes=None, mime_type=None) -> str:
     client = genai.Client(
+        api_key="AIzaSyDolbPUZBPUPvQUu-RGktJmvnUpkcEKIYo",  # replace with env var for safety
     )
     model = "gemini-2.0-flash"
+    parts = []
+    if prompt:
+        parts.append(types.Part.from_text(prompt))
+    if image_bytes:
+        parts.append(types.Part.from_bytes(data=image_bytes, mime_type=mime_type or "image/png"))
+    contents = [types.Content(role="user", parts=parts)]
     config = types.GenerateContentConfig(response_mime_type="text/plain")
     response = client.models.generate_content(
         model=model,
         contents=contents,
         config=config,
     )
     return response.text
 @app.route('/')
 @app.route('/generate', methods=['POST'])
 def gen():
+    prompt = request.form.get("text", "")
+    file = request.files.get("image")
+    image_bytes = None
+    mime_type = None
+    if file:
+        image_bytes = file.read()
+        mime_type = file.mimetype
+    if not prompt and not image_bytes:
+        return jsonify({"error": "No input provided"}), 400
     try:
+        output = generate_from_gemini(prompt, image_bytes, mime_type)
         return jsonify({"result": output})
     except Exception as e:
         app.logger.exception("Generation failed")