Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-import pollinations
+import pollinations as pl
 import pandas as pd
 import os
 from datetime import datetime
@@ -7,11 +7,16 @@ from huggingface_hub import HfApi
 import pyarrow.parquet as pq
 import pyarrow as pa
 import requests
-import json
 
-# Initialize Pollinations text model (default)
+# Initialize Pollinations text model
 default_model = "openai"
-model = pollinations.Text(model=default_model, contextual=True)
+model = pl.Text(
+    model=default_model,
+    system="You are a helpful AI assistant.",
+    contextual=True,
+    seed="random",
+    reasoning_effort="medium"
+)
 
 # Hugging Face setup
 HF_TOKEN = os.getenv("HF_TOKEN")  # Set in HF Space secrets
@@ -33,18 +38,32 @@ def fetch_models():
 
 def change_model(selected_model):
     global model
-    model = pollinations.Text(model=selected_model, contextual=True)
+    model = pl.Text(
+        model=selected_model,
+        system="You are a helpful AI assistant.",
+        contextual=True,
+        seed="random",
+        reasoning_effort="medium"
+    )
     return f"Switched to model: {selected_model}"
 
 def chatbot_response(user_message, history, selected_model):
     global conversation_history, model
     # Ensure model is up-to-date
     if model.model != selected_model:
-        model = pollinations.Text(model=selected_model, contextual=True)
+        model = pl.Text(
+            model=selected_model,
+            system="You are a helpful AI assistant.",
+            contextual=True,
+            seed="random",
+            reasoning_effort="medium"
+        )
     
-    # Generate response
-    seed = int(datetime.now().timestamp())  # Use timestamp as seed
-    response = model(user_message, seed=seed)
+    # Generate response with streaming
+    seed = int(datetime.now().timestamp())
+    response = ""
+    for token in model(user_message, stream=True, seed=seed):
+        response += token
     
     # Append to history with timestamp and model info
     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
@@ -121,4 +140,4 @@ with gr.Blocks() as demo:
     clear.click(lambda: ([], []), None, [chatbot, msg], queue=False)
 
 # Launch the demo
-demo.launch()
+demo.queue().launch()