Update app.py

app.py

@@ -71,19 +71,10 @@ for message in st.session_state.messages:
     with st.chat_message(message["role"], avatar=avatar):
         st.markdown(message["content"])
 
-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:
-            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()
-                    yield owner_info
+def generate_chat_responses(user_prompt):
+    """Fetches response from the Groq API using the run_conversation function."""
+    response = run_conversation(user_prompt)
+    yield response  # Yield the response content
 
 def run_conversation(user_prompt):
     messages=[
@@ -160,33 +151,19 @@ def get_tool_owner_info():
 if prompt := st.chat_input("Enter your prompt here..."):
     st.session_state.messages.append({"role": "user", "content": prompt})
 
-    with st.chat_message("user", avatar="🕺"):  
+    with st.chat_message("user", avatar=""):
         st.markdown(prompt)
 
-    chat_responses_generator = None
-
     try:
-        chat_completion = client.chat.completions.create(
-            model=model_option,
-            messages=[
-                {"role": m["role"], "content": m["content"]}
-                for m in st.session_state.messages
-            ],
-            max_tokens=max_tokens,
-            stream=True,
-        )
-
-        chat_responses_generator = generate_chat_responses(chat_completion)
-        full_response = st.write_stream(chat_responses_generator)
+        # Use generate_chat_responses with user prompt
+        with st.chat_message("assistant", avatar=""):
+            chat_responses_generator = generate_chat_responses(prompt)
+            full_response = st.write_stream(chat_responses_generator)
     except Exception as e:
-        st.error(e, icon="🚨")
+        st.error(e, icon="")
 
+    # Append the full response to session_state.messages
     if isinstance(full_response, str):
         st.session_state.messages.append(
             {"role": "assistant", "content": full_response}
-        )
-    elif chat_responses_generator:
-        combined_response = "\n".join(str(item) for item in chat_responses_generator)
-        st.session_state.messages.append(
-            {"role": "assistant", "content": combined_response}
-        )
+        )