Update app.py

app.py

@@ -9,7 +9,7 @@ client = Groq(api_key=api_key)
 # Initialize conversation history
 conversation_history = []
 
-def chat_with_bot_stream(user_input, temperature, top_p):
+def chat_with_bot_stream(user_input):
     global conversation_history
     conversation_history.append({"role": "user", "content": user_input})
     
@@ -22,9 +22,9 @@ def chat_with_bot_stream(user_input, temperature, top_p):
     completion = client.chat.completions.create(
         model="llama3-70b-8192",
         messages=conversation_history,
-        temperature=temperature,
-        top_p=top_p,
+        temperature=1,
         max_tokens=1024,
+        top_p=1,
         stream=True,
         stop=None,
     )
@@ -35,10 +35,12 @@ def chat_with_bot_stream(user_input, temperature, top_p):
     
     conversation_history.append({"role": "assistant", "content": response_content})
     
-    return conversation_history
+    return [(msg["content"] if msg["role"] == "user" else None, 
+             msg["content"] if msg["role"] == "assistant" else None) 
+            for msg in conversation_history]
 
 # Function to generate a storyboard
-def generate_storyboard(scenario, temperature, top_p):
+def generate_storyboard(scenario):
     if not scenario.strip():
         return "Please provide a scenario to generate the storyboard."
     
@@ -50,9 +52,9 @@ def generate_storyboard(scenario, temperature, top_p):
     completion = client.chat.completions.create(
         model="llama3-70b-8192",
         messages=messages,
-        temperature=temperature,
-        top_p=top_p,
+        temperature=1,
         max_tokens=1024,
+        top_p=1,
         stream=False,
         stop=None,
     )
@@ -65,45 +67,11 @@ h1 { text-align: center; font-size: 24px; margin-bottom: 10px; }
 <h1>📖 Storyboard Assistant</h1>
 """
 
-example_scenarios = [
-    "A futuristic cityscape under AI governance.",
-    "A detective solving a mystery in a cyberpunk world.",
-    "A young explorer discovering an ancient civilization.",
-    "A spaceship crew encountering an unknown planet.",
-    "A medieval knight navigating political intrigue."
-]
-
-example_questions = [
-    "How do I create a compelling storyboard?",
-    "What are the key elements of a good visual story?",
-    "How can AI help with storyboarding?",
-    "What are common mistakes in storyboarding?",
-    "How do I structure a scene effectively?"
-]
-
-temperature_component = gr.Slider(
-    minimum=0,
-    maximum=1,
-    value=1,
-    step=0.01,
-    label="Temperature",
-    info="Controls randomness. Lower values make responses more deterministic."
-)
-
-top_p_component = gr.Slider(
-    minimum=0,
-    maximum=1,
-    value=1,
-    step=0.01,
-    label="Top-P Sampling",
-    info="Limits token selection to tokens with a cumulative probability up to P."
-)
-
 with gr.Blocks(theme=gr.themes.Glass(primary_hue="violet", secondary_hue="emerald", neutral_hue="stone")) as demo:
     with gr.Tabs():
         with gr.TabItem("💬Chat"):
             gr.HTML(TITLE)
-            chatbot = gr.Chatbot(label="Storyboard Chatbot", type="messages")
+            chatbot = gr.Chatbot(label="Storyboard Chatbot")
             with gr.Row():
                 user_input = gr.Textbox(
                     label="Your Message",
@@ -112,25 +80,10 @@ with gr.Blocks(theme=gr.themes.Glass(primary_hue="violet", secondary_hue="emeral
                 )
                 send_button = gr.Button("✋Ask Question")
             
-            with gr.Accordion("🛠️ Customize Chatbot", open=False):
-                gr.Row([temperature_component, top_p_component])
-            
-            with gr.Accordion("🧪 Example Questions", open=False):
-                example_radio = gr.Radio(
-                    choices=example_questions,
-                    label="Example Queries",
-                    info="Select an example question or enter your own."
-                )
-                example_radio.change(
-                    fn=lambda q: q if q else "No question selected.",
-                    inputs=[example_radio],
-                    outputs=[user_input]
-                )
-            
             # Chatbot functionality
             send_button.click(
                 fn=chat_with_bot_stream,
-                inputs=[user_input, temperature_component, top_p_component],
+                inputs=user_input,
                 outputs=chatbot,
                 queue=True
             ).then(
@@ -142,27 +95,8 @@ with gr.Blocks(theme=gr.themes.Glass(primary_hue="violet", secondary_hue="emeral
         with gr.TabItem("📖 Generate Storyboard"):
             gr.Markdown("## Generate a Storyboard")
             scenario_input = gr.Textbox(label="Enter your scenario")
-            example_radio = gr.Radio(
-                choices=example_scenarios,
-                label="Example Scenarios",
-                info="Select an example scenario or enter your own."
-            )
             generate_btn = gr.Button("Generate Storyboard")
             storyboard_output = gr.Textbox(label="Generated Storyboard", interactive=False)
-            
-            with gr.Accordion("🛠️ Customize Storyboard", open=False):
-                gr.Row([temperature_component, top_p_component])
-            
-            generate_btn.click(
-                generate_storyboard, 
-                inputs=[scenario_input, temperature_component, top_p_component], 
-                outputs=storyboard_output
-            )
-            
-            example_radio.change(
-                fn=lambda scenario: scenario if scenario else "No scenario selected.",
-                inputs=[example_radio],
-                outputs=[scenario_input]
-            )
+            generate_btn.click(generate_storyboard, inputs=scenario_input, outputs=storyboard_output)
 
 demo.launch()