Update app.py

app.py

@@ -18,28 +18,36 @@ def make_call(data):
     language = items['lang']
     query = items['text']
     query = query.lower()
-    answer = None
+    translated = None  # Initialize translated to None, it will hold the final output message
 
     # Use the GenerativeModel class to interact with Gemini
     model = genai.GenerativeModel('gemini-pro')  # You can choose a different Gemini model
 
-    while True:
-        try:
-            # Craft your prompt specific for Gemini and desired output format
-            prompt_query = f"Answer this query in a short message with wisdom, love, and compassion, in context to Bhagavad Gita, that feels like chatting to a person and provide references of shlokas from chapters of Bhagavad Gita which are relevant to the query. Keep the answer short, precise, and simple. Query: {query}"
-
-            # Generate content using Gemini
-            response = model.generate_content(prompt_query)
-
-            # Access the generated text from the response
-            answer = response.text  # The generated content is in the 'text' attribute
-
-            # Translate the answer
-            translated = GoogleTranslator(source='auto', target=language).translate(answer)
-        except Exception as e:
-            print(f"API call failed for: {e}")
-        if answer:
-            break
+    try:
+        # Craft your prompt specific for Gemini and desired output format
+        prompt_query = (
+            f"Answer this query in a short message with wisdom, love, and compassion, "
+            f"in context to Bhagavad Gita, that feels like chatting to a person and "
+            f"provide references of shlokas from chapters of Bhagavad Gita which are "
+            f"relevant to the query. Keep the answer short, precise, and simple. "
+            f"Query: {query}"
+        )
+
+        # Generate content using Gemini
+        response = model.generate_content(prompt_query)
+
+        # Access the generated text from the response
+        answer = response.text  # The generated content is in the 'text' attribute
+
+        # Translate the answer
+        translated = GoogleTranslator(source='auto', target=language).translate(answer)
+    except Exception as e:
+        print(f"API call failed for: {e}")
+        # Provide an informative message to the user in case of an error
+        translated = f"An error occurred while fetching the answer: {e}" 
+        # Optionally, you might want to log the full traceback for debugging:
+        # import traceback
+        # traceback.print_exc()
 
     respo = {
         "message": translated,
@@ -54,6 +62,7 @@ gradio_interface.launch()
 
 
 
+
 # import os
 # import gradio as gr
 # from groq import Groq