Update app.py
Browse files
app.py
CHANGED
@@ -70,22 +70,6 @@ rag_chain = (
|
|
70 |
|
71 |
import gradio as gr
|
72 |
|
73 |
-
# Mock implementation of `rag_chain.stream` (replace with your actual implementation)
|
74 |
-
class MockRAGChain:
|
75 |
-
@staticmethod
|
76 |
-
def stream(message):
|
77 |
-
# Mock response stream
|
78 |
-
responses = [
|
79 |
-
f"Processing your message: '{message}'...",
|
80 |
-
"Analyzing your preferences...",
|
81 |
-
"Here are some recommendations for you!"
|
82 |
-
]
|
83 |
-
for response in responses:
|
84 |
-
yield response # Simulate streaming by yielding responses
|
85 |
-
|
86 |
-
# Replace with the actual RAG chain if available
|
87 |
-
rag_chain = MockRAGChain()
|
88 |
-
|
89 |
# Function for the chatbot response stream
|
90 |
def rag_memory_stream(message, history):
|
91 |
partial_text = ""
|
@@ -123,7 +107,7 @@ with gr.Blocks(theme=custom_theme) as demo:
|
|
123 |
with gr.Tabs():
|
124 |
# Chat Tab
|
125 |
with gr.Tab("Chat"):
|
126 |
-
gr.ChatInterface(
|
127 |
fn=rag_memory_stream,
|
128 |
type="messages",
|
129 |
examples=examples,
|
@@ -189,4 +173,4 @@ with gr.Blocks(theme=custom_theme) as demo:
|
|
189 |
|
190 |
# Launch the app
|
191 |
if __name__ == "__main__":
|
192 |
-
demo.launch()
|
|
|
70 |
|
71 |
import gradio as gr
|
72 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
73 |
# Function for the chatbot response stream
|
74 |
def rag_memory_stream(message, history):
|
75 |
partial_text = ""
|
|
|
107 |
with gr.Tabs():
|
108 |
# Chat Tab
|
109 |
with gr.Tab("Chat"):
|
110 |
+
chat_interface = gr.ChatInterface(
|
111 |
fn=rag_memory_stream,
|
112 |
type="messages",
|
113 |
examples=examples,
|
|
|
173 |
|
174 |
# Launch the app
|
175 |
if __name__ == "__main__":
|
176 |
+
demo.launch()
|