Update app.py

app.py

@@ -13,7 +13,7 @@ system_message = {
 
 # Function to reset the chat
 def reset_chat():
-    return [], "New Chat"  # Returns an empty chat history and a new title
+    return [], "New Chat"
 
 # Function to handle the questionnaire submission
 def submit_questionnaire(name, age, location, gender, ethnicity, height, weight,
@@ -87,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 Questionnaire")
+            submit_btn = gr.Button("Submit Inputs")
             reset_btn = gr.Button("Reset Chat")
 
     # Questionnaire with fashion-related questions
@@ -95,16 +95,62 @@ 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"])
 
-    # Chat functionality
-    chatbox = gr.Chatbot(type='messages')
-    user_input = gr.Textbox(label="Your Message", placeholder="Type your message here...")
+    # 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?", ["Very Satisfied", "Satisfied", "Neutral", "Unsatisfied"]),
+        ("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"]),
+        ("Which clothing materials do you prefer?", ["Cotton", "Silk", "Denim", "Synthetic", "Wool"]),
+        ("How important is sustainability in your fashion choices?", ["Very Important", "Somewhat Important", "Not Important"]),
+        ("Do you prefer shopping online or in physical stores?", ["Online", "Physical Stores"]),
+        ("How often do you update your wardrobe?", ["Seasonally", "Every Few Months", "Once a Year", "Rarely"]),
+        ("Do you participate in clothing swaps or second-hand shopping?", ["Yes", "No"]),
+    ]
+
+    # Create a button for filling the questionnaire
+    fill_questionnaire_btn = gr.Button("Fill Questionnaire")
+
+    # Output message for questionnaire submission
+    questionnaire_output = gr.Textbox(label="Questionnaire Submission", interactive=False)
+
+    # Function to collect all questionnaire responses
+    def collect_questionnaire_responses():
+        # Store questionnaire responses in a DataFrame
+        questionnaire_data = {
+            "Style Preference": style_preference.value,
+            "Color Palette": color_palette.value,
+            "Everyday Style": everyday_style.value
+        }
+
+        # Append additional responses
+        for question, choices in fashion_questions:
+            questionnaire_data[question] = gr.Radio(label=question, choices=choices).value
+
+        df = pd.DataFrame([questionnaire_data])  # Create DataFrame from dictionary
+
+        # Append to CSV file
+        df.to_csv("questionnaire_responses.csv", mode='a', header=not pd.io.common.file_exists("questionnaire_responses.csv"), index=False)
+
+        return "Thank you for completing the questionnaire!"
 
     # Connect the buttons to their respective functions
-    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)
+                                                    style_preference, color_palette, everyday_style], outputs=questionnaire_output)
+
+    fill_questionnaire_btn.click(collect_questionnaire_responses, outputs=questionnaire_output)
+
+    reset_btn.click(reset_chat, outputs=[gr.Chatbot(type='messages'), "title"])
+    
+    user_input = gr.Textbox(label="Your Message", placeholder="Type your message here...")
+    chatbox = gr.Chatbot(type='messages')
 
-    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