Update app.py

app.py

@@ -2,23 +2,13 @@ import gradio as gr
 import bcrypt
 from simple_salesforce import Salesforce
 from datetime import datetime
-import openai  # if using Hugging Face-based model (replace with your Hugging Face API call logic)
+from transformers import pipeline  # Hugging Face's transformer library
 
 # Salesforce Connection
 sf = Salesforce(username='diggavalli98@gmail.com', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')
 
-# Hugging Face API Integration (example with GPT for generating offers)
-openai.api_key = "your-huggingface-api-key"  # Replace with your Hugging Face API key
-
-def generate_offer(customer_name):
-    # Example: Use Hugging Face to generate an offer for the customer
-    prompt = f"Generate a personalized offer for {customer_name} for their meal order."
-    response = openai.Completion.create(
-        engine="text-davinci-003",  # Using GPT-3 for the example, replace with Hugging Face model if needed
-        prompt=prompt,
-        max_tokens=50
-    )
-    return response.choices[0].text.strip()
+# Load Hugging Face model for text generation (example: GPT-2 or T5 model)
+generator = pipeline('text-generation', model='gpt2')  # You can replace with a model of your choice
 
 # Function to Hash Password
 def hash_password(password):
@@ -31,7 +21,7 @@ def verify_password(plain_password, hashed_password):
 # Function to handle order and store it in Salesforce
 def save_order_to_salesforce(order_details):
     try:
-        # Use Hugging Face or any other service to generate offers or any AI processing
+        # Use Hugging Face to generate an offer for the customer
         offer = generate_offer(order_details['Customer_Name__c'])
 
         # Creating order data to save in Salesforce
@@ -66,6 +56,12 @@ def save_order_to_salesforce(order_details):
     except Exception as e:
         return f"Error while saving order to Salesforce: {str(e)}"
 
+# Generate a personalized offer using Hugging Face model (you can replace with your own model)
+def generate_offer(customer_name):
+    prompt = f"Generate a personalized discount offer for {customer_name} for their meal order."
+    result = generator(prompt, max_length=50, num_return_sequences=1)
+    offer = result[0]['generated_text']
+    return offer
 
 # Signup and Login Functions
 def signup(name, email, phone, password):
@@ -144,7 +140,7 @@ with gr.Blocks() as app:
     login_button.click(
         lambda email, password: (gr.update(visible=False), gr.update(visible=True), gr.update(value="Menu loaded"), "Login successful!")
         if login(email, password)[0] == "Login successful!" else (gr.update(), gr.update(), gr.update(), "Invalid email or password."),
-        [login_email, login_password], [login_page, menu_page, login_output]
+        [login_email, login_password], [login_page, menu_page, menu_output]
     )
     submit_signup.click(signup, [signup_name, signup_email, signup_phone, signup_password], signup_output)
     preference.change(lambda pref: filter_menu(pref), [preference], menu_output)