Update app.py

app.py

@@ -95,20 +95,27 @@ if prompt := st.chat_input("Type your message..."):
             # Dynamically construct the API URL based on the selected model
             api_url = f"https://api-inference.huggingface.co/models/{selected_model}"
             logger.info(f"Selected model: {selected_model}, API URL: {api_url}")
-            print("payload",payload)
+
             # Query the Hugging Face API using the selected model
             output = query(payload, api_url)
 
             # Handle API response
             if output is not None and isinstance(output, list) and len(output) > 0:
                 if 'generated_text' in output[0]:
-                    assistant_response = output[0]['generated_text']
-                    logger.info(f"Generated response: {assistant_response}")
+                    # Extract the assistant's response
+                    assistant_response = output[0]['generated_text'].strip()
+
+                    # Check for and remove duplicate responses
+                    responses = assistant_response.split("\n</think>\n")
+                    unique_response = responses[0].strip()
+
+                    logger.info(f"Generated response: {unique_response}")
 
+                    # Append response to chat only once
                     with st.chat_message("assistant"):
-                        st.markdown(assistant_response)
+                        st.markdown(unique_response)
 
-                    st.session_state.messages.append({"role": "assistant", "content": assistant_response})
+                    st.session_state.messages.append({"role": "assistant", "content": unique_response})
                 else:
                     logger.error(f"Unexpected API response structure: {output}")
                     st.error("Error: Unexpected response from the model. Please try again.")
@@ -118,4 +125,4 @@ if prompt := st.chat_input("Type your message..."):
 
     except Exception as e:
         logger.error(f"Application Error: {str(e)}", exc_info=True)
-        st.error(f"Application Error: {str(e)}")
+        st.error(f"Application Error: {str(e)}")