Update app.py

app.py

@@ -1,67 +1,40 @@
 import streamlit as st
 import os
 import requests
-from fastapi import FastAPI, Request
-from fastapi.responses import StreamingResponse
-import uvicorn
-from threading import Thread
 
-# --- FASTAPI PROXY LOGIC ---
-# This part handles the API traffic without asking for a password
-api_app = FastAPI()
-
-OLLAMA_URL = "http://localhost:11434"
-
-@api_app.api_route("/api/{path:path}", methods=["GET", "POST", "PUT", "DELETE"])
-async def proxy_ollama(path: Request, path_name: str):
-    url = f"{OLLAMA_URL}/api/{path_name}"
-    method = path.method
-    contents = await path.body()
-    headers = dict(path.headers)
-    
-    # Remove the host header to avoid confusion
-    headers.pop("host", None)
-
-    resp = requests.request(method, url, data=contents, headers=headers, stream=True)
+st.set_page_config(page_title="Core Interface", page_icon="⚡")
+
+# Premium look
+st.markdown("""
+    <style>
+    .main { background-color: #0e1117; color: #ffffff; }
+    .stTextInput > div > div > input { background-color: #262730; color: white; }
+    </style>
+    """, unsafe_allow_html=True)
+
+PIN = os.getenv("PIN", "1234")
+
+if "auth" not in st.session_state:
+    st.session_state.auth = False
+
+if not st.session_state.auth:
+    st.title("🔒 System Restricted")
+    val = st.text_input("Access Key", type="password")
+    if st.button("Unlock"):
+        if val == PIN:
+            st.session_state.auth = True
+            st.rerun()
+        else:
+            st.error("Incorrect.")
+else:
+    st.title("⚡ Server Node Active")
+    st.success("API Path: `/api/generate` (Unlocked)")
+    st.info("Model: `qwen2.5:0.5b` pre-loaded.")
     
-    return StreamingResponse(
-        resp.iter_content(chunk_size=1024),
-        status_code=resp.status_code,
-        headers=dict(resp.headers)
-    )
-
-def run_api():
-    uvicorn.run(api_app, host="0.0.0.0", port=7860)
-
-# --- STREAMLIT UI LOGIC ---
-# This part handles the visual interface on a hidden port
-def run_ui():
-    st.set_page_config(page_title="Core Interface", page_icon="⚡")
-    
-    # Simple CSS for your premium look
-    st.markdown("<style>.main { background-color: #121212; color: #e0e0e0; }</style>", unsafe_allow_html=True)
-
-    SECRET_PIN = os.getenv("PIN", "1234")
-    
-    if "authenticated" not in st.session_state:
-        st.session_state["authenticated"] = False
-
-    if not st.session_state["authenticated"]:
-        st.title("System Locked")
-        user_input = st.text_input("Enter Access Key", type="password")
-        if st.button("Authenticate"):
-            if user_input == SECRET_PIN:
-                st.session_state["authenticated"] = True
-                st.rerun()
-            else:
-                st.error("Access Denied.")
-    else:
-        st.title("⚡ Node Status: Online")
-        st.success("API is public at /api/generate")
-        st.info("Ollama is running on backend port 11434")
-
-# This is a bit of a trick: when HF loads port 7860, 
-# it will see the FastAPI app. We will use a sub-route for the UI.
-if __name__ == "__main__":
-    # If you are running this via streamlit, it executes the UI
-    run_ui()
+    # Status Check
+    if st.button("Check Ollama Health"):
+        try:
+            r = requests.get("http://localhost:11434/api/tags")
+            st.json(r.json())
+        except:
+            st.error("Backend offline.")