Update app.py

app.py

@@ -16,15 +16,14 @@ def translate_text(tamil_text):
     return translation
 
 def query_gemini_api(translated_text):
-    gemini_api_key = os.getenv('GEMINI_API_KEY')
-    url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash-latest:generateContent?key={gemini_api_key}"
+   url = "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash-latest:generateContent"
     headers = {"Content-Type": "application/json"}
+    prompt = f"Based on the following sentence, continue the story: {translated_text}"
     payload = {
-        "contents": [{"parts": [{"text": translated_text}]}]
+        "contents": [{"parts": [{"text": prompt}]}]
     }
-    
-    response = requests.post(url, headers=headers, json=payload)
-    
+    response = requests.post(f"{url}, headers=headers, json=payload)
+
     if response.status_code == 200:
         result = response.json()
         creative_text = result['candidates'][0]['content']['parts'][0]['text']
@@ -32,6 +31,7 @@ def query_gemini_api(translated_text):
     else:
         return f"Error: {response.status_code} - {response.text}"
 
+
 def query_image(payload):
     huggingface_api_key = os.getenv('HUGGINGFACE_API_KEY')
     API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"