SatyamPrakash09 commited on
Commit
f800cd4
·
verified ·
1 Parent(s): 975f51a

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +13 -47
app.py CHANGED
@@ -1,68 +1,34 @@
1
- # Install dependencies (Hugging Face will run this in requirements.txt ideally)
2
- # !pip install langchain langchain-community langchain-google-genai gradio huggingface_hub python-dotenv
3
-
4
  import os
5
  import gradio as gr
6
- from dotenv import load_dotenv
7
-
8
- # Updated imports for latest LangChain (Aug 2025)
9
- from langchain.chains import LLMChain
10
- from langchain_core.prompts import PromptTemplate
11
  from langchain.memory import ConversationBufferMemory
12
- from langchain_google_genai import ChatGoogleGenerativeAI
13
-
14
- # Load API keys from environment
15
- load_dotenv()
16
- os.environ["GOOGLE_API_KEY"] = os.getenv("GOOGLE_API_KEY")
17
 
18
- # Try creating a Gemini Flash 2.5 model
19
- try:
20
- llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash", temperature=0)
21
- response = llm.invoke("Hello Gemini, can you hear me?")
22
- print("✅ API is working!")
23
- print("Response:", response.content)
24
- except Exception as e:
25
- print("❌ API Error:", str(e))
26
-
27
- # Prompt Template
28
- template = """As an adventurous and globetrotting college student,
29
- you're constantly on the lookout for new cultures, experiences, and breathtaking landscapes.
30
- You've visited numerous countries, immersing yourself in local traditions, and you're always eager to swap travel stories
31
- and offer tips on exciting destinations.
32
 
 
33
  {chat_history}
34
  User: {user_message}
35
  Chatbot:"""
36
 
37
  prompt = PromptTemplate(
38
- input_variables=["chat_history", "user_message"],
39
- template=template
40
  )
41
 
42
- # Memory to store chat history
43
  memory = ConversationBufferMemory(memory_key="chat_history")
44
 
45
- # LLM Chain
46
  llm_chain = LLMChain(
47
- llm=llm,
48
  prompt=prompt,
49
- memory=memory
 
50
  )
51
 
52
- # Function for Gradio
53
- def get_text_response(user_message, history):
54
- return llm_chain.predict(user_message=user_message)
55
 
56
- # Gradio Chat Interface
57
- demo = gr.ChatInterface(
58
- fn=get_text_response,
59
- examples=[
60
- "How are you doing?",
61
- "What are your interests?",
62
- "Which places do you like to visit?"
63
- ],
64
- type="messages"
65
- )
66
 
67
  if __name__ == "__main__":
68
- demo.launch()
 
 
 
 
1
  import os
2
  import gradio as gr
3
+ from langchain.chat_models import ChatOpenAI
4
+ from langchain import LLMChain, PromptTemplate
 
 
 
5
  from langchain.memory import ConversationBufferMemory
 
 
 
 
 
6
 
7
+ OPENAI_API_KEY=os.getenv('OPENAI_API_KEY')
 
 
 
 
 
 
 
 
 
 
 
 
 
8
 
9
+ template = """You are a helpful assistant to answer all user queries.
10
  {chat_history}
11
  User: {user_message}
12
  Chatbot:"""
13
 
14
  prompt = PromptTemplate(
15
+ input_variables=["chat_history", "user_message"], template=template
 
16
  )
17
 
 
18
  memory = ConversationBufferMemory(memory_key="chat_history")
19
 
 
20
  llm_chain = LLMChain(
21
+ llm=ChatOpenAI(temperature='0.5', model_name="gpt-3.5-turbo"),
22
  prompt=prompt,
23
+ verbose=True,
24
+ memory=memory,
25
  )
26
 
27
+ def get_text_response(user_message,history):
28
+ response = llm_chain.predict(user_message = user_message)
29
+ return response
30
 
31
+ demo = gr.ChatInterface(get_text_response)
 
 
 
 
 
 
 
 
 
32
 
33
  if __name__ == "__main__":
34
+ demo.launch() #To create a public link, set `share=True` in `launch()`. To enable errors and logs, set `debug=True` in `launch()`.