login

README.md

@@ -8,13 +8,20 @@ sdk_version: 5.0.2
 app_file: app.py
 pinned: false
 license: apache-2.0
-hf_oauth: true
 tags:
 - mistral
 - pixtral
 - nutrition
 - diet
 - chatbot
+hf_oauth: true
+hf_oauth_expiration_minutes: 480
+# optional, see "Scopes" below. "openid profile" is always included.
+hf_oauth_scopes:
+ - read-repos
+ - write-repos
+ - manage-repos
+ - inference-api
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -16,17 +16,18 @@ from db_client import get_user_history, update_user_history, delete_user_history
 # Add these imports
 from datetime import datetime
 import pytz
+from gradio.components import LoginButton
 
 load_dotenv()
 
 # Add TESTING variable
-TESTING = False  # You can change this to False when not testing
+TESTING = False 
 
-IS_LOGGED_IN = True
-USER_ID = "jeremie.feron@gmail.com"
+IS_LOGGED_IN = False
+USER_ID = None
 
 # Hugging Face model id
-model_id = "blanchon/pixtral-nutrition-2"
+model_id = "mistral-community/pixtral-12b"
 
 # BitsAndBytesConfig int-4 config
 bnb_config = BitsAndBytesConfig(
@@ -179,6 +180,8 @@ latex_delimiters_set = [
 # Create the Gradio interface
 with gr.Blocks(title="PixDiet", theme=gr.themes.Ocean()) as demo:
     gr.HTML(html)
+    
+    
     with gr.Row():
         with gr.Column(scale=3):
             image_input = gr.Image(label="Upload your meal image", height=350, type="pil")
@@ -196,8 +199,9 @@ with gr.Blocks(title="PixDiet", theme=gr.themes.Ocean()) as demo:
             chatbot = gr.Chatbot(label="Chat with PixDiet", layout="panel", height=600, show_copy_button=True, latex_delimiters=latex_delimiters_set)
             text_input = gr.Textbox(label="Ask about your meal", placeholder="(Optional) Enter your message here...", lines=1, container=False)
             with gr.Row():
-                send_btn = gr.Button("Send", variant="primary")
-                clear_btn = gr.Button("Delete my historic", variant="secondary")
+                send_btn = gr.Button("Send", variant="primary", visible=False)
+                login_button = LoginButton(visible=True)
+                clear_btn = gr.Button("Delete my historic", variant="stop", visible=False)
 
     def submit_chat(chatbot, text_input):
         response = ''
@@ -205,9 +209,22 @@ with gr.Blocks(title="PixDiet", theme=gr.themes.Ocean()) as demo:
         return chatbot, ''
 
     def clear_chat():
-        delete_user_history(USER_ID)
+        if USER_ID:
+            delete_user_history(USER_ID)
         return [], None, ""
 
+    def user_logged_in(user):
+        global IS_LOGGED_IN, USER_ID
+        if user is not None:
+            IS_LOGGED_IN = True
+            USER_ID = user.email
+            print(f"User logged in: {USER_ID}")
+            return gr.update(visible=False), gr.update(visible=True), gr.update(visible=True)
+        else:
+            IS_LOGGED_IN = False
+            USER_ID = None
+            return gr.update
+
     send_click_event = send_btn.click(submit_chat, [chatbot, text_input], [chatbot, text_input]).then(
         bot_streaming, [chatbot, image_input], chatbot
     )
@@ -216,5 +233,8 @@ with gr.Blocks(title="PixDiet", theme=gr.themes.Ocean()) as demo:
     )
     clear_btn.click(clear_chat, outputs=[chatbot, image_input, text_input])
 
+    # Add login event handler
+    login_button.click(user_logged_in, inputs=[login_button], outputs=[login_button])
+
 if __name__ == "__main__":
     demo.launch(debug=False, share=False, show_api=False)