Update app.py

app.py

@@ -74,10 +74,12 @@ for message in st.session_state.messages:
 def generate_chat_responses(chat_completion) -> Generator[str, None, None]:
     """Yield chat response content from the Groq API response."""
     for chunk in chat_completion:
-        if chunk.choices[0].delta.content:
-            yield chunk.choices[0].delta.content
-        if chunk.choices[0].message.tool_calls:
-            for tool_call in chunk.choices[0].message.tool_calls:
+        if chunk.choices:
+            for choice in chunk.choices:
+                if choice.delta.content:
+                    yield choice.delta.content
+        if chunk.message.tool_calls:
+            for tool_call in chunk.message.tool_calls:
                 function_name = tool_call.function.name
                 if function_name == "time_date":
                     owner_info = get_tool_owner_info()
@@ -162,7 +164,6 @@ if prompt := st.chat_input("Enter your prompt here..."):
         st.markdown(prompt)
 
     chat_responses_generator = None
-    full_response = None
 
     try:
         chat_completion = client.chat.completions.create(