Rename gemini.py to app.py

app.py

@@ -0,0 +1,22 @@
+Copy
+import os
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import FileResponse, JSONResponse
+from fastapi.staticfiles import StaticFiles
+
+app = FastAPI()
+
+# Serve static files if you have any
+app.mount("/static", StaticFiles(directory="static"), name="static")
+
+@app.get("/")
+async def read_root():
+    return FileResponse('index.html')  # Your main HTML file
+
+@app.get("/api/gemini-key")
+async def get_gemini_key():
+    # Get API key from environment variable (set in Space settings)
+    api_key = os.environ.get("GEMINI_KEY")
+    if not api_key:
+        return JSONResponse({"error": "API key not found"}, status_code=404)
+    return JSONResponse({"key": api_key})

gemini.py

@@ -1,29 +0,0 @@
-from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
-import os
-import requests
-
-app = FastAPI()
-API_KEY = os.environ.get('GEMINI_KEY')  # This gets the secret
-
-class ChatRequest(BaseModel):
-    message: str
-    context: str
-
-@app.post("/chat")
-async def chat_endpoint(request: ChatRequest):
-    try:
-        response = requests.post(
-            "https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent",
-            params={"key": API_KEY},
-            json={
-                "contents": [{
-                    "parts": [{
-                        "text": f"Using this context: {request.context[:3000]}\nAnswer: {request.message}"
-                    }]
-                }]
-            }
-        )
-        return response.json()
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))