Upload app.py

app.py

@@ -2,19 +2,17 @@ import gradio as gr
 import pandas as pd
 import os
 import requests
-from huggingface_hub import InferenceClient
 
-# Get Hugging Face token securely
 HF_TOKEN = os.getenv("HF_TOKEN")
 MODEL_ID = "deepset/tinyroberta-squad2"
-
 file_data = {"text": None}
 headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 
 # Handle file upload
 def upload_file(file):
     if file is None or not hasattr(file, "name"):
-        return "❌ No file uploaded."
+        return [], "No file uploaded."
+
     filename = file.name
     try:
         if filename.endswith(".csv"):
@@ -23,18 +21,17 @@ def upload_file(file):
         elif filename.endswith(".txt"):
             file_data["text"] = file.read().decode("utf-8")
         else:
-            return "Please upload a .txt or .csv file."
-        return "File uploaded! Now ask your question."
+            return [], "Please upload a .txt or .csv file."
+        return [], "File uploaded. Now ask your question."
     except Exception as e:
-        return f"Error reading file: {e}"
+        return [], f"Error reading file: {e}"
 
-# Handle user questions
+# Handle user question
 def ask_question(message, history):
     if not message or not file_data["text"]:
-        return "📄 Please upload a file and type a question before asking."
+        return "", history + [[message, "Please upload a file and ask a question."]]
 
-    # Safe now because we passed the checks
-    context = file_data["text"][:1500]  
+    context = file_data["text"][:1500]
 
     payload = {
         "inputs": {
@@ -54,18 +51,17 @@ def ask_question(message, history):
     except Exception as e:
         answer = f"API error: {e}"
 
-    history.append((message, answer))
-    return "", history
-
+    return "", history + [[message, answer]]
 
 # Gradio UI
 with gr.Blocks() as demo:
-    chatbot = gr.Chatbot(label="Chat with Your File")
+    chatbot = gr.Chatbot()
     file_input = gr.File(label="Upload .txt or .csv")
     msg = gr.Textbox(label="Your Question")
     send = gr.Button("Send")
+    status = gr.Textbox(label="Status Message", interactive=False)
 
-    file_input.change(upload_file, inputs=file_input, outputs=chatbot)
+    file_input.change(upload_file, inputs=file_input, outputs=[chatbot, status])
     send.click(ask_question, inputs=[msg, chatbot], outputs=[msg, chatbot])
     gr.ClearButton([msg, file_input, chatbot])