Update app.py

app.py

@@ -20,10 +20,12 @@ if "messages" not in st.session_state:
 # Sidebar configuration
 with st.sidebar:
     st.header("Model Configuration")
+    st.markdown("[Get HuggingFace Token](https://huggingface.co/settings/tokens)")
 
     # Dropdown to select model
     model_options = [
         "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",
+
     ]
     selected_model = st.selectbox("Select Model", model_options, index=0)
 
@@ -53,13 +55,8 @@ def query(payload, api_url):
     headers = {"Authorization": f"Bearer {st.secrets['HF_TOKEN']}"}
     logger.info(f"Sending request to {api_url} with payload: {payload}")
     response = requests.post(api_url, headers=headers, json=payload)
-    try:
-        response.raise_for_status()
-        logger.info(f"Received response: {response.status_code}, {response.text}")
-        return response.json()
-    except requests.exceptions.RequestException as e:
-        logger.error(f"HTTP Request failed: {e}")
-        return None
+    logger.info(f"Received response: {response.status_code}, {response.text}")
+    return response.json()
 
 # Chat interface
 st.title("🤖 DeepSeek Chatbot")
@@ -81,23 +78,15 @@ if prompt := st.chat_input("Type your message..."):
         with st.spinner("Generating response..."):
             # Prepare the payload for the API
             payload = {
-                "inputs": {
-                    "past_user_inputs": [msg["content"] for msg in st.session_state.messages if msg["role"] == "user"],
-                    "generated_responses": [msg["content"] for msg in st.session_state.messages if msg["role"] == "assistant"],
-                    "text": prompt
-                },
+                "inputs": prompt,
                 "parameters": {
                     "max_new_tokens": max_tokens,
                     "temperature": temperature,
                     "top_p": top_p,
                     "return_full_text": False
-                },
-                "system_prompt": system_message
+                }
             }
 
-            # Log the full message parsed to the API
-            logger.info(f"Full message sent to API: {payload}")
-
             # 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}")
@@ -106,7 +95,7 @@ if prompt := st.chat_input("Type your message..."):
             output = query(payload, api_url)
 
             # Handle API response
-            if output and isinstance(output, list) and len(output) > 0 and 'generated_text' in output[0]:
+            if isinstance(output, list) and len(output) > 0 and 'generated_text' in output[0]:
                 assistant_response = output[0]['generated_text']
                 logger.info(f"Generated response: {assistant_response}")
 
@@ -115,9 +104,9 @@ if prompt := st.chat_input("Type your message..."):
 
                 st.session_state.messages.append({"role": "assistant", "content": assistant_response})
             else:
-                logger.error(f"Unexpected API response or error: {output}")
+                logger.error(f"Unexpected API response: {output}")
                 st.error("Error: Unable to generate a response. Please try again.")
 
     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)}")