Update app.py

app.py

@@ -1,52 +1,52 @@
-import streamlit as st
-import requests
-import json
-
-# Set up Streamlit UI
-st.title("DeepSeek-R1 Chat")
-
-# Ollama API endpoint (Ensure Ollama is running on this port)
-OLLAMA_API_URL = "http://localhost:7860/api/generate"
-
-# User input
-prompt = st.text_area("Enter your prompt:")
-
-# Function to call the Ollama API with streaming support
-def generate_response(prompt, stream=True):
-    data = {
-        "model": "deepseek-r1:7b",  # Keep the same model
-        "prompt": prompt,
-        "stream": stream  # Enable streaming
-    }
-
-    try:
-        with requests.post(OLLAMA_API_URL, json=data, headers={"Content-Type": "application/json"}, stream=stream) as response:
-            response.raise_for_status()
-            if stream:
-                for chunk in response.iter_lines(decode_unicode=True):
-                    if chunk:
-                        chunk_data = json.loads(chunk)
-                        yield chunk_data.get("response", "")
-            else:
-                response_data = response.json()
-                yield response_data.get("response", "No response received.")
-    except requests.exceptions.RequestException as e:
-        yield f"Error: {e}"
-
-string = ''
-for chunk in generate_response('hello!', stream=True):
-    string += chunk
-
-# Button to trigger response
-if st.button("Generate Response"):
-    if prompt.strip():
-        st.subheader("Ollama API Response:")
-        response_placeholder = st.empty()  # Placeholder for streaming output
-
-        full_response = ""
-        for chunk in generate_response(prompt, stream=True):
-            full_response += chunk
-            response_placeholder.markdown(full_response)
-
-    else:
-        st.warning("Please enter a prompt.")
+# import streamlit as st
+# import requests
+# import json
+
+# # Set up Streamlit UI
+# st.title("DeepSeek-R1 Chat")
+
+# # Ollama API endpoint (Ensure Ollama is running on this port)
+# OLLAMA_API_URL = "http://localhost:7860/api/generate"
+
+# # User input
+# prompt = st.text_area("Enter your prompt:")
+
+# # Function to call the Ollama API with streaming support
+# def generate_response(prompt, stream=True):
+#     data = {
+#         "model": "deepseek-r1:7b",  # Keep the same model
+#         "prompt": prompt,
+#         "stream": stream  # Enable streaming
+#     }
+
+#     try:
+#         with requests.post(OLLAMA_API_URL, json=data, headers={"Content-Type": "application/json"}, stream=stream) as response:
+#             response.raise_for_status()
+#             if stream:
+#                 for chunk in response.iter_lines(decode_unicode=True):
+#                     if chunk:
+#                         chunk_data = json.loads(chunk)
+#                         yield chunk_data.get("response", "")
+#             else:
+#                 response_data = response.json()
+#                 yield response_data.get("response", "No response received.")
+#     except requests.exceptions.RequestException as e:
+#         yield f"Error: {e}"
+
+# # string = ''
+# # for chunk in generate_response('hello!', stream=True):
+# #     string += chunk
+
+# # Button to trigger response
+# if st.button("Generate Response"):
+#     if prompt.strip():
+#         st.subheader("Ollama API Response:")
+#         response_placeholder = st.empty()  # Placeholder for streaming output
+
+#         full_response = ""
+#         for chunk in generate_response(prompt, stream=True):
+#             full_response += chunk
+#             response_placeholder.markdown(full_response)
+
+#     else:
+#         st.warning("Please enter a prompt.")