Update app.py

app.py

@@ -160,9 +160,14 @@ def generate_response(inputs):
     ]
     # Generate the response using the pipeline
     response = pipe(messages)
-    # Extract the generated text from the response and clean it up if necessary
-    generated_text = response[0]["generated_text"] if response else "No response generated"
-    return generated_text.strip()  # Removing any unwanted whitespace
+    # Extract the generated assistant's response
+    try:
+        # Navigate the nested structure to find the assistant's response content
+        assistant_response = response[0]["generated_text"][1]["content"]
+        return assistant_response.strip()  # Clean up any extra whitespace
+    except (IndexError, KeyError, TypeError) as e:
+        # Handle any errors due to unexpected structure in the response
+        return "Error processing response. Please try again."
 
 # Antarmuka Gradio
 iface = gr.Interface(