Update app.py

app.py

@@ -1,66 +1,86 @@
+```python
 import gradio as gr
-from huggingface_hub import InferenceClient
-
-
-"""
-For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
-"""
-client = InferenceClient("meta-llama/Llama-3.1-8B-Instruct")
-
-
-def respond(
-    message,
-    history: list[tuple[str, str]],
-    system_message,
-    max_tokens,
-    temperature,
-    top_p,
-):
-    
-
-    messages = [{"role": "system", "content": system_message}]
-
-    for val in history:
-        if val[0]:
-            messages.append({"role": "user", "content": val[0]})
-        if val[1]:
-            messages.append({"role": "assistant", "content": val[1]})
-
-    messages.append({"role": "user", "content": message})
-
-    response = ""
-
-    for message in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        token = message.choices[0].delta.content
-
-        response += token
-        yield response
-
-"""
-For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
-"""
-demo = gr.ChatInterface(
-    respond,
-    additional_inputs=[
-        gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-        gr.Slider(
-            minimum=0.1,
-            maximum=1.0,
-            value=0.95,
-            step=0.05,
-            label="Top-p (nucleus sampling)",
-        ),
-    ],
-)
-
-
-if __name__ == "__main__":
-    demo.launch()
+import random
+import time
+def chatbot(message, history, system_prompt):
+    # Simple response generation based on the system prompt and user message
+    if not system_prompt:
+        system_prompt = "I am a helpful AI assistant."
+    # Simulate typing with a stream
+    for i in range(1, 11):
+        thinking_time = random.uniform(0.1, 0.3)
+        time.sleep(thinking_time)
+        # Generate response gradually based on system prompt
+        if "friendly" in system_prompt.lower():
+            response_parts = [
+                "Hi there! ", 
+                "I'd be happy to help you with that. ",
+                f"You asked about '{message}'. ",
+                "Based on what I know, I think I can provide some guidance on this topic."
+            ]
+        elif "expert" in system_prompt.lower():
+            response_parts = [
+                "Based on my expertise, ",
+                f"regarding '{message}', ",
+                "I can provide you with a detailed analysis. ",
+                "Here's what you should know about this subject from a professional perspective."
+            ]
+        elif "funny" in system_prompt.lower():
+            response_parts = [
+                "Well, well, well! ", 
+                f"So you want to know about '{message}'? ",
+                "That's quite an interesting question! ",
+                "Let me share my thoughts with a touch of humor."
+            ]
+        else:
+            response_parts = [
+                "Thank you for your question. ", 
+                f"Regarding '{message}', ",
+                "I can provide the following information: ",
+                "Please let me know if you need any clarification or have follow-up questions."
+            ]
+        partial_response = "".join(response_parts[:int(i * len(response_parts) / 10)])
+        yield partial_response
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🤖 Custom AI Chatbot")
+    gr.Markdown("Configure your chatbot's personality and knowledge using the system prompt.")
+    with gr.Row():
+        with gr.Column(scale=2):
+            chatbot = gr.ChatInterface(
+                fn=chatbot,
+                additional_inputs=[
+                    gr.Textbox(
+                        placeholder="Enter system prompt here...",
+                        label="System Prompt",
+                        info="Define your chatbot's personality, knowledge, and style",
+                        lines=5
+                    )
+                ],
+                examples=[
+                    ["What's the weather like today?"],
+                    ["Can you tell me about quantum physics?"],
+                    ["Write a short poem about nature."]
+                ],
+                title="Chat with Your Custom AI",
+            )
+        with gr.Column(scale=1):
+            gr.Markdown("## System Prompt Examples")
+            system_prompt_examples = gr.Examples(
+                examples=[
+                    "You are a friendly assistant who speaks casually and uses simple language.",
+                    "You are an expert in technology with deep knowledge of AI, machine learning, and computer science.",
+                    "You are a funny comedian who responds with humor and wit.",
+                    "You are a professional business consultant providing concise, valuable advice.",
+                    "You are a supportive tutor who explains complex concepts in simple terms."
+                ],
+                inputs=chatbot.additional_inputs[0],
+                label="Click on an example to use it"
+            )
+            gr.Markdown("""
+            ### Tips for System Prompts:
+            - Define personality traits
+            - Specify knowledge domains
+            - Set the tone and speaking style
+            - Include specific instructions
+            - Add any constraints or limitations
+            """)