Update app.py

app.py

@@ -12,16 +12,16 @@ openai.api_key = "your_openai_api_key_here"  # Replace with your OpenAI API key
 
 # Function to generate the initial answer with BLIP and expand it with OpenAI API
 def qna(image, question):
-    # Step 1: Get initial short answer from BLIP
-    inputs = processor(image, question, return_tensors="pt")
-    out = model.generate(**inputs)
-    short_answer = processor.decode(out[0], skip_special_tokens=True)
+    try:
+        # Step 1: Get initial short answer from BLIP
+        inputs = processor(image, question, return_tensors="pt")
+        out = model.generate(**inputs)
+        short_answer = processor.decode(out[0], skip_special_tokens=True)
 
-    # Step 2: Construct prompt for OpenAI API
-    prompt = f"Question: {question}\nShort Answer: {short_answer}\nProvide a detailed explanation based on this answer."
+        # Step 2: Construct prompt for OpenAI API
+        prompt = f"Question: {question}\nShort Answer: {short_answer}\nProvide a detailed explanation based on this answer."
 
-    # Step 3: Send prompt to OpenAI API for a paragraph-length answer
-    try:
+        # Step 3: Send prompt to OpenAI API for a paragraph-length answer
         response = openai.Completion.create(
             engine="text-davinci-003",  # Specify model
             prompt=prompt,
@@ -29,18 +29,20 @@ def qna(image, question):
         )
         detailed_answer = response.choices[0].text.strip()
 
+        return detailed_answer
+
     except openai.error.OpenAIError as e:
-        # Print detailed error information
+        # Log full error details in the backend
         print(f"OpenAI API error: {e}")
-        detailed_answer = f"Failed to get response from OpenAI API: {e}"
+        # Return a user-friendly error message to the Gradio interface
+        return "An error occurred while generating the response. Please try again later."
 
     except Exception as e:
+        # Log any other exceptions in the backend
         print(f"General exception: {e}")
-        detailed_answer = "Failed to connect to OpenAI API."
-
-    return detailed_answer
+        # Return a general error message to the Gradio interface
+        return "A technical issue occurred. Please try again later."
 
 # Create Gradio interface
 interf = gr.Interface(fn=qna, inputs=["image", "text"], outputs="text")
 interf.launch()
-