Update app.py

app.py

@@ -1,42 +1,3 @@
-# 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
 
@@ -67,8 +28,8 @@ iface = gr.Interface(
     fn=extract_entities,
     inputs=gr.Textbox(label="Enter your text"),
     outputs=gr.Markdown(label="Extracted Entities"),
-    title="Entity Extraction Tool",
-    description="Extract entities from text using Gemini 1.5 Pro."
+    title="Entity Extraction Tool for Email",
+    description="Extract entities from Email using Gemini 1.5 Pro."
 )