Update app.py

app.py

@@ -1,21 +1,26 @@
-import os
+import streamlit as st
 import requests
-import gradio as gr
 
-# Define function to query the AI model
+# Set up Hugging Face API details
+headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+API_URL = "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"
+
+# Function to query the Hugging Face model
 def query(payload):
-    token = os.getenv("HF_TOKEN", None)
-    headers = {"Authorization": f"Bearer {token}"}
-    API_URL = "https://api-inference.huggingface.co/models/facebook/bart-large-cnn"
     response = requests.post(API_URL, headers=headers, json=payload)
     return response.json()
 
-# Create Gradio interface
-def chat(input_text):
-    data = query({"inputs": input_text, "parameters": {"do_sample": False}})
-    return data[0]['generated_text']
-
-input_text = gr.Textbox(lines=7, label="Input Text")
-output_text = gr.Textbox(label="Model Response")
+# Streamlit UI
+st.title("Chat App with Hugging Face")
+user_input = st.text_input("You:", "")
 
-gr.Interface(fn=chat, inputs=input_text, outputs=output_text, title="AI Chat", description="Chat with AI model").launch()
+if st.button("Send"):
+    if user_input.strip() != "":
+        # Query Hugging Face model
+        data = query({"inputs": user_input, "parameters": {"do_sample": False}})
+        
+        # Display response
+        if data and "summary_text" in data[0]:
+            st.text_area("Bot:", value=data[0]["summary_text"], height=150)
+        else:
+            st.error("No response from the model")