Update app.py

app.py

@@ -15,14 +15,15 @@ def translate_text(tamil_text):
     translation = tokenizer.decode(translated_tokens[0], skip_special_tokens=True)
     return translation
 
-def query_gemini_api(translated_text):
+def query_gemini_api(translated_text, 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": prompt}]}]
     }
-    response = requests.post(url, headers=headers, json=payload)  # Corrected this line
+    response = requests.post(f"{url}?key={gemini_api_key}", headers=headers, json=payload)
+
     if response.status_code == 200:
         result = response.json()
         creative_text = result['candidates'][0]['content']['parts'][0]['text']
@@ -38,12 +39,14 @@ def query_image(payload):
     return response.content
 
 def process_input(tamil_input):
+    gemini_api_key = os.getenv('GEMINI_API_KEY')  
     translated_output = translate_text(tamil_input)
-    creative_output = query_gemini_api(translated_output)
+    creative_output = query_gemini_api(translated_output, gemini_api_key)
     image_bytes = query_image({"inputs": translated_output})
     image = Image.open(io.BytesIO(image_bytes))
     return translated_output, creative_output, image
 
+
 iface = gr.Interface(
     fn=process_input,
     inputs=[gr.Textbox(label="Input Tamil Text")],