Update app.py

app.py

@@ -19,15 +19,17 @@ def translate_text(tamil_text):
     translation = tokenizer.decode(translated_tokens[0], skip_special_tokens=True)
     return translation
 
-def query_gemini_api(translated_text):
-    url = "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-turbo:generateText"
+def query_gemini_api(translated_text, gemini_api_key, max_output_tokens=300):
+    url = f"https://generativelanguage.googleapis.com/v1beta2/models/gemini-1.5:generateText?key={gemini_api_key}"
     headers = {"Content-Type": "application/json"}
     payload = {
         "prompt": {"text": translated_text},
-        "temperature": 0.7
+        "temperature": 0.7,
+        "max_output_tokens": max_output_tokens
     }
-    response = requests.post(f"{url}?key={gemini_api_key}", headers=headers, json=payload)
-
+    
+    response = requests.post(url, headers=headers, json=payload)
+    
     if response.status_code == 200:
         result = response.json()
         creative_text = result['candidates'][0]['output']