Update project_model.py

project_model.py

@@ -198,8 +198,23 @@ def process_inputs(
     # Send prompt to Gemma 3
     gemma_output = gemma_pipe(text=session.message_history, max_new_tokens=200)
 
+    # Debugging: Check the output structure
+    print("Gemma Output:", gemma_output)
+    print("Type of Gemma Output:", type(gemma_output))
+
+    # Make sure the output is in the expected format (a string)
+    if isinstance(gemma_output, list) and len(gemma_output) > 0:
+        # Assuming it's a list of dicts with the 'generated_text' field
+        gemma_text = gemma_output[0].get("generated_text", "")
+        if isinstance(gemma_text, str):
+            answer = gemma_text
+        else:
+            # Handle unexpected formats or empty text
+            answer = "No valid generated text found."
+    else:
+        answer = "No valid output from Gemma model"
     
-    answer = gemma_output[0]["generated_text"][-1]["content"]
+    # answer = gemma_output[0]["generated_text"][-1]["content"]
 
     # Append GEMMA's response to the history to maintain alternating structure
     session.add_answer(answer)