Raveheart1 commited on
Commit
453fd37
·
verified ·
1 Parent(s): 902b2db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -17,18 +17,17 @@ def translate_text(tamil_text):
17
 
18
  def query_gemini_api(translated_text):
19
  gemini_api_key = os.getenv('GEMINI_API_KEY')
20
- url = f"https://generativelanguage.googleapis.com/v1beta2/models/gemini-1.5-flash:generateText?key={gemini_api_key}"
21
  headers = {"Content-Type": "application/json"}
22
  payload = {
23
- "prompt": {"text": translated_text},
24
- "temperature": 0.7
25
  }
26
 
27
  response = requests.post(url, headers=headers, json=payload)
28
 
29
  if response.status_code == 200:
30
  result = response.json()
31
- creative_text = result['candidates'][0]['output']
32
  return creative_text
33
  else:
34
  return f"Error: {response.status_code} - {response.text}"
@@ -58,4 +57,5 @@ iface = gr.Interface(
58
  title="TRANSART🎨 BY Sakthi",
59
  description="Enter Tamil text to translate to English and generate an image based on the translated text."
60
  )
 
61
  iface.launch()
 
17
 
18
  def query_gemini_api(translated_text):
19
  gemini_api_key = os.getenv('GEMINI_API_KEY')
20
+ url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash-latest:generateContent?key={gemini_api_key}"
21
  headers = {"Content-Type": "application/json"}
22
  payload = {
23
+ "contents": [{"parts": [{"text": translated_text}]}]
 
24
  }
25
 
26
  response = requests.post(url, headers=headers, json=payload)
27
 
28
  if response.status_code == 200:
29
  result = response.json()
30
+ creative_text = result['candidates'][0]['content']['parts'][0]['text']
31
  return creative_text
32
  else:
33
  return f"Error: {response.status_code} - {response.text}"
 
57
  title="TRANSART🎨 BY Sakthi",
58
  description="Enter Tamil text to translate to English and generate an image based on the translated text."
59
  )
60
+
61
  iface.launch()