Update app.py

app.py

@@ -1,16 +1,51 @@
+# import gradio as gr
+# import os
+
+# from langchain_google_genai import GoogleGenerativeAI
+# from langchain import LLMChain
+# from langchain.prompts import ChatPromptTemplate,PromptTemplate,HumanMessagePromptTemplate
+# from langchain.schema import (
+#     HumanMessage,
+#     SystemMessage
+# )
+
+# google_api_key=os.environ["google_api_key"]
+
+
+# llm = GoogleGenerativeAI(model='gemini-1.5-pro', google_api_key=google_api_key)
+
+# messages = [
+#     SystemMessage(content="You are an expert at extracting entities from text."),
+#     HumanMessagePromptTemplate.from_template("{text}")
+# ]
+
+# chat_prompt = ChatPromptTemplate.from_messages(messages)
+# chain = LLMChain(llm=llm, prompt=chat_prompt)
+
+# def extract_entities(text):
+#     result = chain.run(text)
+#     return result
+
+# def chatbot(text, history):
+#     response = extract_entities(text)
+#     return response
+
+# iface = gr.ChatInterface(
+#     fn=chatbot,
+#     title="Entity Extraction Chatbot",
+#     description="Extract entities from text using Gemini 1.5 Pro."
+# )
+
+# iface.launch()
 import gradio as gr
 import os
 
 from langchain_google_genai import GoogleGenerativeAI
 from langchain import LLMChain
-from langchain.prompts import ChatPromptTemplate,PromptTemplate,HumanMessagePromptTemplate
-from langchain.schema import (
-    HumanMessage,
-    SystemMessage
-)
-
-google_api_key=os.environ["google_api_key"]
+from langchain.prompts import ChatPromptTemplate, HumanMessagePromptTemplate
+from langchain.schema import SystemMessage
 
+google_api_key = os.getenv("google_api_key")
 
 llm = GoogleGenerativeAI(model='gemini-1.5-pro', google_api_key=google_api_key)
 
@@ -20,20 +55,21 @@ messages = [
 ]
 
 chat_prompt = ChatPromptTemplate.from_messages(messages)
+
 chain = LLMChain(llm=llm, prompt=chat_prompt)
 
 def extract_entities(text):
     result = chain.run(text)
     return result
 
-def chatbot(text, history):
-    response = extract_entities(text)
-    return response
-
-iface = gr.ChatInterface(
-    fn=chatbot,
-    title="Entity Extraction Chatbot",
+iface = gr.Interface(
+    fn=extract_entities,
+    inputs=gr.Textbox(label="Enter your text"),
+    outputs=gr.Textbox(label="Extracted Entities"),
+    title="Entity Extraction Tool",
     description="Extract entities from text using Gemini 1.5 Pro."
 )
 
+
 iface.launch()
+