Update app.py

app.py

@@ -1,33 +1,36 @@
-import os
-from huggingface_hub import InferenceClient
-import gradio as gr
+from transformers import pipeline
 
-# Function to test API connection
-def test_api_connection(query: str, api_key: str) -> str:
-    """Tests the Hugging Face Inference API connection with the provided API key."""
-    try:
-        # Using a different model for testing
-        client = InferenceClient(model="distilbert-base-uncased", token=api_key)
-        response = client.text_classification(query)
-        return f"API connection successful! Response: {response}"
-    except Exception as e:
-        return f"Error: {str(e)}"
+# Predefined context
+context = """
+The Scrum framework consists of Scrum Teams and their roles, events, and artifacts. 
+Scrum Teams are self-managing and cross-functional. The main roles include the Scrum Master, 
+Product Owner, and Development Team. The Daily Stand-up is a short meeting for the Scrum Team 
+to synchronize activities and create a plan for the next 24 hours.
+"""
 
-# Gradio interface function
-def run_agent(query):
-    api_key = os.getenv("WingTokenAll")
-    if not api_key:
-        return "Error: wingTokenRead not found in environment variables."
-    return test_api_connection(query, api_key)
+# Load the question-answering model
+question_answerer = pipeline("question-answering", model='distilbert-base-cased-distilled-squad')
 
-# Create and launch Gradio interface
-interface = gr.Interface(
-    fn=run_agent,
-    inputs=gr.Textbox(label="Enter a test query", placeholder="e.g., This is a test sentence"),
-    outputs=gr.Textbox(label="API Test Response"),
-    title="Hugging Face API Test",
-    description="Enter a test query to verify the Hugging Face API connection."
-)
+
+def answer_question(question):
+    """Answer the question based on the predefined context."""
+    result = question_answerer(question=question, context=context)
+    return result['answer']
+
+def main():
+    print("Welcome to the Simple Q&A Agent!")
+    print("Type 'exit' to quit.\n")
+    
+    while True:
+        user_input = input("Ask a question: ")
+        
+        if user_input.lower() == 'exit':
+            print("Goodbye!")
+            break
+        
+  
+        answer = answer_question(user_input)
+        print(f"Answer: {answer}\n")
 
 if __name__ == "__main__":
-    interface.launch(server_port=7860, share=True, quiet=True)
+    main()