Update app.py

app.py

@@ -1,86 +1,114 @@
 import gradio as gr
-from modules import api_manager, health_ai, database
+from modules import api_manager, health_ai, database, image_gen
 import os
 
-# Initialize DB on startup
+# Initialize Database on startup
 database.init_db()
 
 # --- LOGIC FUNCTIONS ---
 
+# 1. Menu Analysis Logic
 def process_menu_upload(image, language):
     if image is None:
         return "No image provided", "", {}, "Error"
-
-    # 1. OCR Step
     extracted_text, ocr_source = api_manager.perform_ocr(image)
-    
-    # 2. Translation Step
     translated_text, trans_source = api_manager.translate_text(extracted_text, language)
-    
-    # 3. Health Analysis Step
     health_data = health_ai.analyze_health(translated_text)
-    
-    # Logs
     api_log = f"OCR Source: {ocr_source}\nTranslation Source: {trans_source}"
-    
     return extracted_text, translated_text, health_data, api_log
 
+# 2. 3D Image Generation Logic
+def generate_dish_visualization(dish_name):
+    if not dish_name:
+        return None, "Please enter a dish name."
+    image, status = image_gen.generate_food_image(dish_name)
+    return image, status
+
+# 3. Auth & Database Logic
 def register_rest_ui(name, loc, owner):
     return database.add_restaurant(name, loc, owner)
-
 def search_rest_ui(query):
     return database.search_restaurants(query)
+def signup_ui(email, password, role):
+    return database.create_user(email, password, role)
+def login_ui(email, password):
+    return database.verify_login(email, password)
 
 # --- UI LAYOUT ---
 
 with gr.Blocks(theme=gr.themes.Soft(), title="MenuVision AI") as demo:
     gr.Markdown("# 🍽️ MenuVision AI")
-    gr.Markdown("### Intelligent Restaurant Menu Platform (Dual-API Architecture)")
+    gr.Markdown("### Intelligent Restaurant Menu & 3D Visualization Platform")
 
     with gr.Tabs():
         
-        # TAB 1: MENU ANALYSIS
-        with gr.Tab("📸 Guest: Analyze Menu"):
+        # --- TAB 1: LOGIN & SIGNUP ---
+        with gr.Tab("🔐 Login & Signup"):
+            with gr.Row():
+                with gr.Column():
+                    gr.Markdown("### 📝 Create Account")
+                    su_email = gr.Textbox(label="Email")
+                    su_pass = gr.Textbox(label="Password", type="password")
+                    su_role = gr.Radio(["User", "Restaurant Owner"], label="Account Type", value="User")
+                    su_btn = gr.Button("Sign Up", variant="primary")
+                    su_msg = gr.Textbox(label="Status")
+                    su_btn.click(signup_ui, inputs=[su_email, su_pass, su_role], outputs=su_msg)
+                
+                with gr.Column():
+                    gr.Markdown("### 🔑 Login")
+                    li_email = gr.Textbox(label="Email")
+                    li_pass = gr.Textbox(label="Password", type="password")
+                    li_btn = gr.Button("Login")
+                    li_msg = gr.Textbox(label="Login Status")
+                    li_btn.click(login_ui, inputs=[li_email, li_pass], outputs=li_msg)
+
+        # --- TAB 2: MENU SCANNER ---
+        with gr.Tab("📸 Menu Scanner"):
+            gr.Markdown("Upload a menu to extract text, translate it, and check health stats.")
             with gr.Row():
                 with gr.Column():
                     img_input = gr.Image(type="pil", label="Upload Menu Image")
                     lang_input = gr.Dropdown(
-                        ["es", "fr", "de", "ur", "zh", "ar"], 
-                        label="Translate to", 
-                        value="es"
+                        ["es", "fr", "de", "ur", "zh", "ar", "hi", "ru", "pt"], 
+                        label="Translate to", value="es"
                     )
                     analyze_btn = gr.Button("Analyze Menu", variant="primary")
-                
                 with gr.Column():
-                    ocr_output = gr.Textbox(label="Extracted Text (OCR)", interactive=False, lines=4)
+                    ocr_output = gr.Textbox(label="Extracted Text", interactive=False, lines=4)
                     trans_output = gr.Textbox(label="Translated Text", interactive=False, lines=4)
                     health_output = gr.JSON(label="Health Insights")
-                    api_status = gr.Textbox(label="System Logs", interactive=False)
+                    api_status = gr.Textbox(label="Logs", interactive=False)
+            analyze_btn.click(process_menu_upload, inputs=[img_input, lang_input], outputs=[ocr_output, trans_output, health_output, api_status])
 
-            analyze_btn.click(
-                process_menu_upload, 
-                inputs=[img_input, lang_input], 
-                outputs=[ocr_output, trans_output, health_output, api_status]
-            )
+        # --- TAB 3: 3D FOOD VISUALIZER ---
+        with gr.Tab("🍔 3D Food Visualizer"):
+            gr.Markdown("### See it before you order!")
+            gr.Markdown("Type a dish name to generate a realistic **3D image on a restaurant table**.")
+            
+            with gr.Row():
+                with gr.Column():
+                    dish_input = gr.Textbox(label="Enter Dish Name (e.g., 'Spicy Pepperoni Pizza')")
+                    gen_btn = gr.Button("Generate 3D Image", variant="primary")
+                with gr.Column():
+                    gen_output = gr.Image(label="AI Generated Preview")
+                    gen_status = gr.Textbox(label="Status", interactive=False)
+            
+            gen_btn.click(generate_dish_visualization, inputs=dish_input, outputs=[gen_output, gen_status])
 
-        # TAB 2: SEARCH RESTAURANTS
-        with gr.Tab("🔍 Guest: Search Restaurants"):
-            search_input = gr.Textbox(label="Search by Name or Location (e.g., 'Pizza')")
+        # --- TAB 4: SEARCH ---
+        with gr.Tab("🔍 Search Restaurants"):
+            search_input = gr.Textbox(label="Search by Name or Location")
             search_btn = gr.Button("Search")
             search_output = gr.Dataframe(label="Results")
             search_btn.click(search_rest_ui, inputs=search_input, outputs=search_output)
 
-        # TAB 3: OWNER DASHBOARD
-        with gr.Tab("👨‍🍳 Owner: Register"):
-            gr.Markdown("Add your restaurant to the public database.")
-            with gr.Row():
-                r_name = gr.Textbox(label="Restaurant Name")
-                r_loc = gr.Textbox(label="Location")
-                r_owner = gr.Textbox(label="Owner Name")
-            
+        # --- TAB 5: OWNER DASHBOARD ---
+        with gr.Tab("👨‍🍳 Owner Dashboard"):
+            r_name = gr.Textbox(label="Restaurant Name")
+            r_loc = gr.Textbox(label="Location")
+            r_owner = gr.Textbox(label="Owner Name")
             reg_btn = gr.Button("Register Restaurant")
             reg_msg = gr.Textbox(label="Status")
-            
             reg_btn.click(register_rest_ui, inputs=[r_name, r_loc, r_owner], outputs=reg_msg)
 
 if __name__ == "__main__":