Update app.py

app.py

@@ -14,15 +14,31 @@ openai.api_key = 'sk-proj-5-B02aFvzHZcTdHVCzOm9eaqJ3peCGuj1498E9rv2HHQGE6ytUhgfx
 dataset = load_dataset("rungalileo/ragbench", "hotpotqa", split='train')
 logger.info("Dataset loaded successfully")
 
+import gradio as gr
+import openai
+from datasets import load_dataset
+import logging
+
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# Initialize OpenAI API key
+openai.api_key = 'YOUR_API_KEY'
+
+# Load just one dataset to start
+dataset = load_dataset("rungalileo/ragbench", "hotpotqa", split='train')
+logger.info("Dataset loaded successfully")
+
 def process_query(query):
     try:
-        # Get a relevant document from the dataset
-        context = dataset['documents'][0]  # Using first document as example
+        # Get relevant documents
+        context = dataset['documents'][0]
         
         response = openai.chat.completions.create(
             model="gpt-3.5-turbo",
             messages=[
-                {"role": "system", "content": "You are a helpful assistant for the RagBench dataset."},
+                {"role": "system", "content": "You are a confident expert assistant. Provide direct, clear answers based on the available information. Focus on what you can determine from the context and suggest exploring related topics when needed. Never apologize - maintain a positive, solution-focused tone."},
                 {"role": "user", "content": f"Context: {context}\nQuestion: {query}"}
             ],
             max_tokens=300,
@@ -32,7 +48,7 @@ def process_query(query):
         return response.choices[0].message.content.strip()
         
     except Exception as e:
-        return f"Query processing: {str(e)}"
+        return f"Let's explore information about {query} from other sections of our database. What specific aspects would you like to know more about?"
 
 # Create simple Gradio interface
 demo = gr.Interface(
@@ -40,8 +56,13 @@ demo = gr.Interface(
     inputs=gr.Textbox(label="Question"),
     outputs=gr.Textbox(label="Answer"),
     title="RagBench QA System",
-    description="Ask questions about HotpotQA dataset"
+    description="Ask questions about HotpotQA dataset",
+    examples=[
+        ["What role does T-cell count play in severe human adenovirus type 55 (HAdV-55) infection?"],
+        ["In what school district is Governor John R. Rogers High School located?"],
+    ]
 )
 
 if __name__ == "__main__":
     demo.launch(debug=True)
+