Update app.py

app.py

@@ -5,17 +5,25 @@ from typing import Iterator
 import google.generativeai as genai
 import time # Import time module for potential debugging/delay
 
+print("import library complete")
+print("add API key")
+
 # get Gemini API Key from the environ variable
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 genai.configure(api_key=GEMINI_API_KEY)
 
+print("add API key complete")
+print("add model")
+
 used_model = "gemini-2.0-flash-thinking-exp-01-21"
 
 # we will be using the Gemini 2.0 Flash model with Thinking capabilities
 model = genai.GenerativeModel("gemini-2.0-flash-thinking-exp-01-21")
 
+print(f"add model {used_model} colplete")
 
 def format_chat_history(messages: list) -> list:
+    print("start format history")
     """
     Formats the chat history into a structure Gemini can understand
     """
@@ -31,21 +39,31 @@ def format_chat_history(messages: list) -> list:
         #    })
             
         #print(f"t2 {message}")
-        formatted_history.append({
-            "role": "user" if message.get("role") == "user" else "model",
+
+        if message.get("role") == "user" :
+            formatted_history.append({
+            "role": "user",
             "parts": [message.get("content", "")]
-        })
+            })
+        elif message.get("role") == "model" :
+            formatted_history.append({
+            "role": "model",
+            "parts": [message.get("content", "")]
+            })
         
     #print(f"t3 {formatted_history}")
+    print("return formatted history")
     return formatted_history
 
 def stream_gemini_response(user_message: str, messages: list) -> Iterator[list]:
+    print("start model response stream")
     """
     Streams thoughts and response with conversation history support for text input only.
     """
     if not user_message.strip(): # Robust check: if text message is empty or whitespace
         messages.append(ChatMessage(role="assistant", content="Please provide a non-empty text message. Empty input is not allowed.")) # More specific message
         yield messages
+        print("Empty text message")
         return
 
     try:
@@ -56,7 +74,9 @@ def stream_gemini_response(user_message: str, messages: list) -> Iterator[list]:
         chat_history = format_chat_history(messages)
 
         # Initialize Gemini chat
+        print("Chat parameter")
         chat = model.start_chat(history=chat_history)
+        print("Start response")
         response = chat.send_message(user_message, stream=True)
 
         # Initialize buffers and flags
@@ -132,7 +152,7 @@ def stream_gemini_response(user_message: str, messages: list) -> Iterator[list]:
                     metadata={"title": "⚙️ Thinking: *The thoughts produced by the model are experimental"}
                 )
             #time.sleep(0.05) #Optional: Uncomment this line to add a slight delay for debugging/visualization of streaming. Remove for final version
-
+            print("Response end")
             yield messages
 
         print(f"\n=== Final Response ===\n{response_buffer}")