Create app.py

app.py

@@ -0,0 +1,46 @@
+import streamlit as st
+import requests
+import os
+
+# -----------------------------------------------------
+# Streamlit Page Configuration
+# -----------------------------------------------------
+st.set_page_config(page_title="Ollama API Tester", layout="wide")
+st.title("🧪 Ollama API Model Tester")
+st.subheader("Test Ollama via API in Hugging Face Spaces.")
+
+# -----------------------------------------------------
+# API Key and URL (Replace with Your Ollama Server)
+# -----------------------------------------------------
+OLLAMA_API_URL = "https://your-ollama-server.com/api/chat"  # Change this to your server
+
+# -----------------------------------------------------
+# User Input: Prompt
+# -----------------------------------------------------
+user_prompt = st.text_area("✍️ Enter your prompt:", "Explain quantum computing in simple terms.")
+
+# -----------------------------------------------------
+# Button to Generate Response
+# -----------------------------------------------------
+if st.button("🚀 Generate Response"):
+    with st.spinner("Fetching Ollama response..."):
+        try:
+            # Prepare API payload
+            payload = {
+                "model": "mistral",  # Change model name if needed
+                "messages": [{"role": "user", "content": user_prompt}]
+            }
+
+            # Send request to Ollama API
+            response = requests.post(OLLAMA_API_URL, json=payload)
+
+            # Process Response
+            if response.status_code == 200:
+                response_text = response.json()["message"]["content"]
+                st.markdown("### 🤖 Ollama API Response:")
+                st.write(f"📝 {response_text}")
+            else:
+                st.error(f"❌ Ollama API Error: {response.json()}")
+
+        except Exception as e:
+            st.error(f"❌ Ollama Request Failed: {e}")