Update app.py

app.py

@@ -111,49 +111,38 @@ By adhering to these rules and utilizing the provided examples as guidelines, we
 
 # # Launch the app
 # iface.launch()
-def get_response(prompt, model_choice, chat_history):
+def get_response(prompt, model_choice, chat_history=None):  # Default `chat_history` to None if not provided
     client = Groq(api_key=GROQ_API_KEY)
-    
-    # Append the new user prompt to the chat history
-    if chat_history:
-        chat_history += f"\nUser: {prompt}"
+
+    if chat_history is None:
+        chat_history = "User: " + prompt
     else:
-        chat_history = f"User: {prompt}"
-    
-    # Create a chat completion with the user's question, selected model, and the chat history
+        chat_history += "\nUser: " + prompt
+
     chat_completion = client.chat.completions.create(
-        messages=[
-            {
-                "role": "user",
-                "content": f"You are helpful Interior design AI assistant named Vinci based on the provided caption below please answer the following:{prompt}\n \
-                caption:{SYSTEM_PROMPT_TEMPLATE}, Please keep your answer concise. {chat_history}",
-            }
-        ],
+        messages=[{
+            "role": "user",
+            "content": f"Your prompt: {prompt}\n\nChat History:\n{chat_history}"
+        }],
         model=model_choice,
     )
-    
-    # Extract the model response
-    model_response = chat_completion.choices[0].message.contents
-    
-    # Update the chat history with the model's response
-    updated_chat_history = chat_history + f"\nVinci: {model_response}"
-    
-    # Return the model response and the updated chat history
+
+    model_response = chat_completion.choices[0].message.content
+    updated_chat_history = chat_history + "\nVinci: " + model_response
+
     return model_response, updated_chat_history
 
-# Define the Gradio interface inputs, including a hidden State input for maintaining the chat history
-# Update the Gradio interface definition
 iface = gr.Interface(
     fn=get_response,
     inputs=[
         gr.Textbox(lines=2, placeholder="Enter your question here..."),
-        gr.Radio(choices=["mixtral-8x7b-32768", "llama2-70b-4096"], label="Model Selection", value="mixtral-8x7b-32768"),
-        gr.State()  # This will keep the chat history
+        gr.Radio(choices=["mixtral-8x7b-32768", "llama2-70b-4096"], label="Model Selection"),
+        gr.State()
     ],
     outputs=[
         gr.TextArea(label="Answer"),
-        gr.State()  # This will output the updated chat history
-    ],
-    title="Vinci 2.0 Using Groq Test",
-    description="Hi I am Vinci, your interior design Assistant. Ask any question and select a model to get a response from the Groq chat model."
-)
+        gr.State()
+    ]
+)
+
+iface.launch()