Fixed API response and switched to environment variable

src/streamlit_app.py

@@ -2,21 +2,30 @@ import streamlit as st
 import requests
 import os
 
-st.title("My Hugging Face Model Interface")
+st.title("Uzmi GPT - Romantic Quote Generator")
 
-input_text = st.text_area("Enter your text:", "")
-HF_TOKEN = os.getenv("HF_TOKEN")  # Loaded from Hugging Face Secrets
+# Input prompt
+user_input = st.text_area("Enter your prompt:", "A romantic love quote about forever:")
 
-if st.button("Get Prediction") and input_text:
-    headers = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
-    response = requests.post(
-        "https://api-inference.huggingface.co/models/rajan3208/uzmi-gpt",
-        headers=headers,
-        json={"inputs": input_text}
-    )
+# Button to trigger prediction
+if st.button("Generate"):
+    API_URL = "https://api-inference.huggingface.co/models/rajan3208/uzmi-gpt"
+    headers = {
+        "Authorization": f"Bearer {os.environ.get('HF_TOKEN')}"
+    }
+
+    payload = {"inputs": user_input}
+
+    with st.spinner("Generating..."):
+        response = requests.post(API_URL, headers=headers, json=payload)
 
     if response.status_code == 200:
-        st.success("Prediction:")
-        st.write(response.json()[0]['generated_text'])
+        try:
+            generated_text = response.json()[0]["generated_text"]
+            st.success("Generated Text:")
+            st.write(generated_text)
+        except Exception as e:
+            st.error("Could not parse response.")
+            st.json(response.json())
     else:
         st.error(f"Error {response.status_code}: {response.text}")