Update app.py

app.py

@@ -1,63 +1,21 @@
 import gradio as gr
-from huggingface_hub import InferenceClient
-
-"""
-For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
+from utils import get_answer
+title = "Ask tax bot a Question"
+description = """
+Introducing Tax Bot, a sophisticated chatbot powered by Langchain and the advanced Huggingface Zephyr-7b-beta model. Specialized in Indian tax laws, Tax Bot is ready to answer your queries. Don’t hesitate, ask Tax Bot anything about Indian tax laws!
+<img src="https://github.com/pratikroy311/TAXBot/blob/main/bot_image.jpeg?raw=true" width=200px>
 """
-client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
-
-
-def respond(
-    message,
-    history: list[tuple[str, str]],
-    system_message,
-    max_tokens,
-    temperature,
-    top_p,
-):
-    messages = [{"role": "system", "content": system_message}]
-
-    for val in history:
-        if val[0]:
-            messages.append({"role": "user", "content": val[0]})
-        if val[1]:
-            messages.append({"role": "assistant", "content": val[1]})
-
-    messages.append({"role": "user", "content": message})
-
-    response = ""
-
-    for message in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        token = message.choices[0].delta.content
-
-        response += token
-        yield response
-
-"""
-For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
-"""
-demo = gr.ChatInterface(
-    respond,
-    additional_inputs=[
-        gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-        gr.Slider(
-            minimum=0.1,
-            maximum=1.0,
-            value=0.95,
-            step=0.05,
-            label="Top-p (nucleus sampling)",
-        ),
-    ],
-)
-
 
-if __name__ == "__main__":
-    demo.launch()
+article = "Article coming soon"
+
+gr.Interface(
+    fn=get_answer,
+    inputs="textbox",
+    outputs="text",
+    title=title,
+    description=description,
+    article=article,
+    examples=[
+        ["My mother acquired a plot of land 12 years ago for 8 lakhs. Presently, she intends to sell it for 33 lakhs to purchase a flat. The government’s assigned circle rate for this land stands at 17 lakhs.To prevent overpayment of stamp duty during registration, we’ll register the sale at the circle rate. Now, how can we handle the remaining amount above the circle rate? Can we manage it through cash or online transactions? Moreover, what are the best methods to minimize tax liabilities in this scenario?Also, will my mother need to file her Income Tax Return (ITR) due to this transaction?"],
+        ["What are the income tax rules for a salaried employee having salary less than 10 lakhs?"]],
+    ).launch()