Update app.py

app.py

@@ -1,7 +1,8 @@
 import os
 import datetime
 import re
-import httpx
+import json
+import requests
 from fastapi import FastAPI, Request, Form
 from fastapi.responses import HTMLResponse
 from fastapi.templating import Jinja2Templates
@@ -9,55 +10,44 @@ from fastapi.templating import Jinja2Templates
 app = FastAPI()
 templates = Jinja2Templates(directory="templates")
 
-# Retrieve the Gemini API key from an environment variable
-GEMINI_API_KEY = os.getenv("GEMINI_API_KEY", "AIzaSyDcYyq3w21iwipYn17wCAQo3AYWhUIGDSI")
-# Hypothetical Gemini API endpoint
-GEMINI_API_URL = "https://api.gemini.example/v1/generate"
+# Retrieve your Gemini API key from environment variables
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY", "your_api_key_here")
+GEMINI_API_URL = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key={GEMINI_API_KEY}"
 
-async def call_gemini_api(prompt: str) -> str:
-    """
-    Call the Gemini API with a given prompt and return the generated text.
-    """
+def call_gemini_api(prompt: str) -> str:
     headers = {
-        "Authorization": f"Bearer {GEMINI_API_KEY}",
         "Content-Type": "application/json"
     }
-    payload = {
-        "prompt": prompt,
-        "max_tokens": 60,  # Adjust as needed
-        "temperature": 0.7
+    data = {
+        "contents": [{
+            "parts": [{"text": prompt}]
+        }]
     }
-    async with httpx.AsyncClient() as client:
-        response = await client.post(GEMINI_API_URL, json=payload, headers=headers)
-        response.raise_for_status()
-        data = response.json()
-        # Assume the API returns the generated text in the "text" field
-        return data.get("text", "Keep exploring coding and robotics!")
+    response = requests.post(GEMINI_API_URL, headers=headers, data=json.dumps(data))
+    response_data = response.json()
+    # Assuming the generated text is in the first candidate's "output" field:
+    if "candidates" in response_data and len(response_data["candidates"]) > 0:
+        return response_data["candidates"][0].get("output", "Keep coding and exploring robotics!")
+    return "Keep coding and exploring robotics!"
 
 @app.get("/", response_class=HTMLResponse)
-async def read_index(request: Request):
+def read_index(request: Request):
     return templates.TemplateResponse("index.html", {"request": request})
 
 @app.post("/fortune", response_class=HTMLResponse)
-async def get_fortune(
-    request: Request,
-    birthday: str = Form(...),
-    standard: str = Form(...)
-):
-    # Validate birthday format (YYYY-MM-DD)
+def get_fortune(request: Request, birthday: str = Form(...), standard: str = Form(...)):
     try:
-        birth_date = datetime.datetime.strptime(birthday, "%Y-%m-%d").date()
+        datetime.datetime.strptime(birthday, "%Y-%m-%d").date()
     except Exception:
         return templates.TemplateResponse(
             "index.html",
             {"request": request, "error": "Invalid date format! Please use YYYY-MM-DD."}
         )
 
-    # Extract numeric grade from the standard string (e.g., "1st" -> "1")
     grade_match = re.search(r'\d+', standard)
     grade = grade_match.group(0) if grade_match else "1"
 
-    # Build a prompt for Gemini API to generate a fortune message
+    # Construct a prompt that emphasizes coding and robotics
     prompt = (
         f"You are an inspirational fortune teller for young coders and robotics enthusiasts. "
         f"A student with birthday {birthday} and in grade {standard} is curious about technology. "
@@ -66,16 +56,11 @@ async def get_fortune(
     )
 
     try:
-        fortune_message = await call_gemini_api(prompt)
+        fortune_message = call_gemini_api(prompt)
     except Exception as e:
         fortune_message = "Coding and robotics hold a bright future for you! Keep dreaming and exploring."
 
     return templates.TemplateResponse(
         "index.html",
-        {
-            "request": request,
-            "fortune": fortune_message,
-            "birthday": birthday,
-            "standard": standard
-        }
+        {"request": request, "fortune": fortune_message, "birthday": birthday, "standard": standard}
     )