Update app.py

app.py

@@ -78,36 +78,84 @@ By adhering to these rules and utilizing the provided examples as guidelines, we
 ### Assistant:
 """
 
-def get_response(prompt, model_choice):
-    client = Groq(
-      api_key=GROQ_API_KEY,
-      )
-    # Create a chat completion with the user's question and selected model
+# def get_response(prompt, model_choice):
+#     client = Groq(
+#       api_key=GROQ_API_KEY,
+#       )
+#     # Create a chat completion with the user's question and selected model
+#     chat_completion = client.chat.completions.create(
+#         messages=[
+#             {
+#                 "role": "user",
+#                 "content": f"You are helpful Interior design AI assistant named Vinci based on the provoided caption below please answer the following:{prompt}\n \
+#                 caption:{SYSTEM_PROMPT_TEMPLATE}, Please keep your answer concise",
+#             }
+#         ],
+#         model=model_choice,
+#     )
+
+#     # Extract and return the response
+#     return chat_completion.choices[0].message.content
+
+# # Define the Gradio interface inputs including a Radio component for model selection
+# 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")
+#     ],
+#     outputs=gr.TextArea(label="Answer"),
+#     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."
+# )
+
+# # Launch the app
+# iface.launch()
+def get_response(prompt, model_choice, chat_history):
+    client = Groq(api_key=GROQ_API_KEY)
+    
+    # Append the new user prompt to the chat history
+    if chat_history:
+        chat_history += f"\nUser: {prompt}"
+    else:
+        chat_history = f"User: {prompt}"
+    
+    # Create a chat completion with the user's question, selected model, and the chat history
     chat_completion = client.chat.completions.create(
         messages=[
             {
                 "role": "user",
-                "content": f"You are helpful Interior design AI assistant named Vinci based on the provoided caption below please answer the following:{prompt}\n \
-                caption:{SYSTEM_PROMPT_TEMPLATE}, Please keep your answer concise",
+                "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}",
             }
         ],
         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
+    return model_response, updated_chat_history
 
-    # Extract and return the response
-    return chat_completion.choices[0].message.content
-
-# Define the Gradio interface inputs including a Radio component for model selection
+# Define the Gradio interface inputs, including a hidden State input for maintaining the chat history
 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.Radio(choices=["mixtral-8x7b-32768", "llama2-70b-4096"], label="Model Selection", value="mixtral-8x7b-32768"),
+        gr.State()  # This will keep the chat history
+    ],
+    outputs=[
+        gr.TextArea(label="Answer"),
+        gr.State()  # This will output the updated chat history
     ],
-    outputs=gr.TextArea(label="Answer"),
     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."
 )
 
 # Launch the app
-iface.launch()
+iface.launch()