Update rag_gradio_app.py

rag_gradio_app.py

@@ -23,14 +23,25 @@ def rag_pipeline(question):
     # Generate AI response
     full_prompt = f"Context: {context}\\n\\nQuestion: {question}"
     import openai
+import openai
 
-response = openai.ChatCompletion.create(
-    model="gpt-4",
-    messages=[{"role": "system", "content": "You are a helpful assistant."},
-              {"role": "
+def generate_response(question, context):
+    """Generate AI response using OpenAI GPT-4"""
+    full_prompt = f"Context: {context}\n\nQuestion: {question}"
+    
+    response = openai.ChatCompletion.create(
+        model="gpt-4",
+        messages=[
+            {"role": "system", "content": "You are an AI assistant helping with question answering."},
+            {"role": "user", "content": full_prompt}
+        ],
+        max_tokens=300,
+        temperature=0.7
+    )
+    
+    return response['choices'][0]['message']['content'].strip()
 
     
-    return response['choices'][0]['message']['content'].strip(), retrieved_docs
 
 # Gradio UI
 iface = gr.Interface(