jithenderchoudary commited on
Commit
4c59466
·
verified ·
1 Parent(s): 73ebd8d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -10
app.py CHANGED
@@ -1,22 +1,18 @@
1
  import os
2
  from flask import Flask, render_template, request, jsonify
3
- import openai
4
- import requests
5
  from dotenv import load_dotenv
6
- from utils import create_case_in_salesforce
7
 
8
  # Load environment variables from .env file
9
  load_dotenv()
10
 
11
  # Set up API keys
12
  HUGGINGFACE_API_KEY = os.getenv('HUGGINGFACE_API_KEY')
13
- SALESFORCE_API_URL = os.getenv('SALESFORCE_API_URL')
14
- SALESFORCE_API_KEY = os.getenv('SALESFORCE_API_KEY')
15
 
16
  # Set up Flask app
17
  app = Flask(__name__)
18
 
19
- # Set up Hugging Face API Client
20
  client = InferenceClient(HUGGINGFACE_API_KEY)
21
 
22
  def respond(message, history, system_message, max_tokens, temperature, top_p):
@@ -58,10 +54,6 @@ def chat():
58
  # Get response from Hugging Face model
59
  bot_reply = respond(user_message, history, system_message, max_tokens, temperature, top_p)
60
 
61
- # If the FAQ response requires a case to be created, initiate the case creation
62
- if "create a case" in bot_reply.lower():
63
- create_case_in_salesforce(user_message)
64
-
65
  return jsonify({'response': bot_reply})
66
 
67
  # Home page for the chatbot UI
 
1
  import os
2
  from flask import Flask, render_template, request, jsonify
3
+ from huggingface_hub import InferenceClient # Import InferenceClient correctly
 
4
  from dotenv import load_dotenv
 
5
 
6
  # Load environment variables from .env file
7
  load_dotenv()
8
 
9
  # Set up API keys
10
  HUGGINGFACE_API_KEY = os.getenv('HUGGINGFACE_API_KEY')
 
 
11
 
12
  # Set up Flask app
13
  app = Flask(__name__)
14
 
15
+ # Initialize the Hugging Face API Client
16
  client = InferenceClient(HUGGINGFACE_API_KEY)
17
 
18
  def respond(message, history, system_message, max_tokens, temperature, top_p):
 
54
  # Get response from Hugging Face model
55
  bot_reply = respond(user_message, history, system_message, max_tokens, temperature, top_p)
56
 
 
 
 
 
57
  return jsonify({'response': bot_reply})
58
 
59
  # Home page for the chatbot UI