Update app.py

app.py

@@ -13,7 +13,7 @@ system_message = {
 
 # Function to reset the chat
 def reset_chat():
-    return [], "New Chat"
+    return [], "New Chat"  # Returns an empty chat history and a new title
 
 # Function to handle the questionnaire submission
 def submit_questionnaire(name, age, location, gender, ethnicity, height, weight,
@@ -39,22 +39,6 @@ def submit_questionnaire(name, age, location, gender, ethnicity, height, weight,
 
     return "Thank you for completing the questionnaire!"
 
-# Function to collect responses to additional fashion-related questions
-def collect_questionnaire_responses(style_preference, color_palette, everyday_style):
-    # Store additional responses in a DataFrame
-    additional_data = {
-        "Style Preference": style_preference,
-        "Color Palette": color_palette,
-        "Everyday Style": everyday_style
-    }
-
-    df = pd.DataFrame([additional_data])  # Create DataFrame from dictionary
-
-    # Append to CSV file
-    df.to_csv("additional_questionnaire_responses.csv", mode='a', header=not pd.io.common.file_exists("additional_questionnaire_responses.csv"), index=False)
-
-    return "Thank you for answering the additional questions!"
-
 # Function to handle chat
 def chat(user_input, messages):
     if user_input:
@@ -89,7 +73,7 @@ def chat(user_input, messages):
 
 # Gradio Interface
 with gr.Blocks() as demo:
-    gr.Markdown("## FRIDAY")
+    gr.Markdown("## Fashion Assistant Chatbot")
 
     # Sidebar for user inputs
     with gr.Row():
@@ -103,7 +87,7 @@ with gr.Blocks() as demo:
             weight = gr.Number(label="Weight (kg)", value=70, minimum=20, maximum=200)
 
         with gr.Column():
-            submit_btn = gr.Button("Submit User Inputs")
+            submit_btn = gr.Button("Submit Questionnaire")
             reset_btn = gr.Button("Reset Chat")
 
     # Questionnaire with fashion-related questions
@@ -111,54 +95,16 @@ with gr.Blocks() as demo:
     color_palette = gr.Radio(label="What color palette do you wear often?", choices=["Neutrals", "Bright Colors", "Pastels", "Dark Shades"])
     everyday_style = gr.Radio(label="How would you describe your everyday style?", choices=["Relaxed", "Trendy", "Elegant", "Bold"])
 
-    # Additional fashion-related questions
-    fashion_questions = [
-        ("What do you prioritize when choosing an outfit?", ["Comfort", "Style", "Affordability", "Brand"]),
-        ("How often do you experiment with new trends?", ["Always", "Sometimes", "Rarely", "Never"]),
-        ("What kind of accessories do you usually wear?", ["Watches", "Rings", "Necklaces", "Bracelets", "Earrings"]),
-        ("Do you follow fashion trends?", ["Always", "Sometimes", "Never"]),
-        ("How satisfied are you with your wardrobe?", ["Yes", "No"]),
-        ("Do you consider your style unique?", ["Yes", "No"]),
-        ("How confident do you feel in your style?", ["Very Confident", "Somewhat Confident", "Not Confident"]),
-        ("Where do you look for fashion inspiration?", ["Social Media", "Fashion Magazines", "Friends", "Other"]),
-        ("Do you have specific attire for special occasions?", ["Yes", "No"]),
-        ("Do you wear gender-neutral clothing?", ["Yes", "No"]),
-        ("What influences your fashion choices the most?", ["Celebrities", "Social Media", "Friends", "Personal Preference"]),
-        ("Do you enjoy shopping for clothes?", ["Yes", "No"]),
-        ("How often do you declutter your wardrobe?", ["Regularly", "Occasionally", "Rarely"]),
-        ("Do you have a favorite fashion brand?", ["Yes", "No"]),
-        ("How do you feel about thrift shopping?", ["Love it", "Neutral", "Dislike it"]),
-    ]
-
-    # Create a separate button for the questionnaire
-    fill_questionnaire_btn = gr.Button("Fill Additional Questionnaire")
-    
-    # Adding additional questions
-    additional_responses = []
-    for question, choices in fashion_questions:
-        additional_responses.append(gr.Radio(label=question, choices=choices))
-
     # Chat functionality
     chatbox = gr.Chatbot(type='messages')
     user_input = gr.Textbox(label="Your Message", placeholder="Type your message here...")
 
     # Connect the buttons to their respective functions
-    submit_btn.click(
-        submit_questionnaire,
-        inputs=[name, age, location, gender, ethnicity, height, weight,
-                style_preference, color_palette, everyday_style],
-        outputs=gr.Textbox(label="Submission Response")
-    )
-    
-    fill_questionnaire_btn.click(
-        collect_questionnaire_responses,
-        inputs=[style_preference, color_palette, everyday_style],
-        outputs=gr.Textbox(label="Additional Questions Response")
-    )
-
-    # Corrected reset button output
-    reset_btn.click(reset_chat, outputs=[chatbox, "title"])
+    output_message = gr.Textbox(label="Output Message")  # Define an output component
+    submit_btn.click(submit_questionnaire, inputs=[name, age, location, gender, ethnicity, height, weight,
+                                                    style_preference, color_palette, everyday_style], outputs=output_message)
 
+    reset_btn.click(reset_chat, outputs=[chatbox, output_message])  # Corrected outputs
     user_input.submit(chat, inputs=[user_input, chatbox], outputs=[chatbox, user_input])
 
 # Run the app