Update app.py

app.py

@@ -10,7 +10,7 @@ from fastapi.templating import Jinja2Templates
 app = FastAPI()
 templates = Jinja2Templates(directory="templates")
 
-# Retrieve your Gemini API key from environment variables
+# Get your Gemini API key from environment variables
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY", "AIzaSyDcYyq3w21iwipYn17wCAQo3AYWhUIGDSI")
 GEMINI_API_URL = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key={GEMINI_API_KEY}"
 
@@ -23,44 +23,46 @@ def call_gemini_api(prompt: str) -> str:
             "parts": [{"text": prompt}]
         }]
     }
-    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!"
+    try:
+        response = requests.post(GEMINI_API_URL, headers=headers, data=json.dumps(data))
+        response.raise_for_status()
+        response_data = response.json()
+        # Expecting the API to return candidates with an "output" field.
+        if "candidates" in response_data and len(response_data["candidates"]) > 0:
+            return response_data["candidates"][0].get("output", "Keep coding and stay inspired!")
+        return "Keep coding and stay inspired!"
+    except Exception as e:
+        return "Keep coding and stay inspired!"
 
 @app.get("/", response_class=HTMLResponse)
-def read_index(request: Request):
+def index(request: Request):
     return templates.TemplateResponse("index.html", {"request": request})
 
 @app.post("/fortune", response_class=HTMLResponse)
-def get_fortune(request: Request, birthday: str = Form(...), standard: str = Form(...)):
+def fortune(request: Request, birthday: str = Form(...), standard: str = Form(...)):
     try:
         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."}
+            {"request": request, "error": "Invalid birthday format! Please use YYYY-MM-DD."}
         )
-
-    grade_match = re.search(r'\d+', standard)
-    grade = grade_match.group(0) if grade_match else "1"
-
-    # Construct a prompt that emphasizes coding and robotics
+    
+    # Build a prompt that uses the provided birthday and standard.
     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. "
-        f"Provide an uplifting fortune message that encourages them to explore coding and robotics, "
-        f"emphasizing creativity, learning, and fun in technology."
+        f"You are an inspirational fortune teller for young coders. "
+        f"Here's the birthday: {birthday}. For a student in standard {standard}, "
+        f"provide a random, modern, concise, and energetic quote that inspires them to code and explore robotics."
     )
-
-    try:
-        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."
-
+    
+    fortune_message = call_gemini_api(prompt)
+    
     return templates.TemplateResponse(
         "index.html",
-        {"request": request, "fortune": fortune_message, "birthday": birthday, "standard": standard}
+        {
+            "request": request,
+            "fortune": fortune_message,
+            "birthday": birthday,
+            "standard": standard
+        }
     )