Update app.py

app.py

@@ -45,4 +45,28 @@ def query_faiss(query, index, embedder, chunks, top_k=2):
     return "\n\n".join(retrieved_chunks)
 
 # Build prompt and generate answer
-def chat_with
+def chat_with_transcript(query):
+    context = query_faiss(query, index, embedder, chunks)
+    prompt = f"""You are an AI assistant. Use the following context to answer the question.
+Context:
+{context}
+Question: {query}
+Provide your answer below:
+"""
+    response = llm(prompt)[0]['generated_text']
+    print("Raw model response:", response)  # Debug statement
+    return response.strip()
+
+# Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("# 📄 Chat with a Transcript")
+    query_input = gr.Textbox(label="Ask a question about the transcript")
+    answer_output = gr.Textbox(label="Answer")
+    
+    query_input.submit(
+        chat_with_transcript,
+        inputs=[query_input],
+        outputs=[answer_output]
+    )
+
+demo.launch()