Upload 2 files

Dockerfile

@@ -0,0 +1,16 @@
+# Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
+# you will also find guides on how best to write your Dockerfile
+
+FROM python:3.9
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . /app
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,91 @@
+from fastapi import FastAPI, Request, Form, UploadFile, File
+from fastapi.templating import Jinja2Templates
+from fastapi.responses import HTMLResponse, RedirectResponse
+from fastapi.staticfiles import StaticFiles
+from dotenv import load_dotenv
+import os, io
+from PIL import Image
+import markdown
+import google.generativeai as genai
+
+# Load environment variable
+load_dotenv()
+API_KEY = os.getenv("GOOGLE_API_KEY") or "AIzaSyDXqoZkx92J1jt_XAYxCGEHmYQtg2XwfLU"
+genai.configure(api_key=API_KEY)
+
+app = FastAPI()
+templates = Jinja2Templates(directory="templates")
+app.mount("/static", StaticFiles(directory="static"), name="static")
+
+model = genai.GenerativeModel('gemini-2.0-flash')
+
+# Create a global chat session
+chat = None
+chat_history = []
+
+@app.get("/", response_class=HTMLResponse)
+async def root(request: Request):
+    return templates.TemplateResponse("index.html", {
+        "request": request,
+        "chat_history": chat_history,
+    })
+
+@app.post("/", response_class=HTMLResponse)
+async def handle_input(
+    request: Request,
+    user_input: str = Form(...),
+    image: UploadFile = File(None)
+):
+    global chat, chat_history
+    
+    # Initialize chat session if needed
+    if chat is None:
+        chat = model.start_chat(history=[])
+    
+    parts = []
+    if user_input:
+        parts.append(user_input)
+
+    # For display in the UI
+    user_message = user_input
+    
+    if image and image.content_type.startswith("image/"):
+        data = await image.read()
+        try:
+            img = Image.open(io.BytesIO(data))
+            parts.append(img)
+            user_message += " [Image uploaded]"  # Indicate image in chat history
+        except Exception as e:
+            chat_history.append({
+                "role": "model",
+                "content": markdown.markdown(f"**Error loading image:** {e}")
+            })
+            return RedirectResponse("/", status_code=303)
+
+    # Store user message for display
+    chat_history.append({"role": "user", "content": user_message})
+
+    try:
+        # Send message to Gemini model
+        resp = chat.send_message(parts)
+        # Add model response to history
+        raw = resp.text
+        chat_history.append({"role": "model", "content": raw})
+
+    except Exception as e:
+        err = f"**Error:** {e}"
+        chat_history.append({
+            "role": "model",
+            "content": markdown.markdown(err)
+        })
+
+    # Post-Redirect-Get
+    return RedirectResponse("/", status_code=303)
+
+# Clear chat history and start fresh
+@app.post("/new")
+async def new_chat():
+    global chat, chat_history
+    chat = None
+    chat_history.clear()
+    return RedirectResponse("/", status_code=303)