changes to interface
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ from PyPDF2 import PdfReader
|
|
| 4 |
|
| 5 |
# Models Setup
|
| 6 |
models = {
|
| 7 |
-
"
|
| 8 |
"client": InferenceClient(model="HuggingFaceH4/zephyr-7b-beta"),
|
| 9 |
},
|
| 10 |
"PDF Summarizer (T5)": {
|
|
@@ -22,7 +22,7 @@ def chat_with_model(model_choice, user_message, chat_history, file=None):
|
|
| 22 |
user_message += f"\n\nPDF Content:\n{pdf_text}"
|
| 23 |
|
| 24 |
if not user_message.strip():
|
| 25 |
-
return chat_history
|
| 26 |
|
| 27 |
model_info = models[model_choice]
|
| 28 |
client = model_info["client"]
|
|
@@ -52,7 +52,7 @@ def chat_with_model(model_choice, user_message, chat_history, file=None):
|
|
| 52 |
|
| 53 |
# Update Chat History
|
| 54 |
chat_history.append((user_message, response))
|
| 55 |
-
return chat_history
|
| 56 |
|
| 57 |
# Function to Extract Text from PDF
|
| 58 |
def extract_text_from_pdf(file):
|
|
@@ -82,14 +82,14 @@ def create_chat_interface():
|
|
| 82 |
with gr.Row():
|
| 83 |
model_choice = gr.Dropdown(
|
| 84 |
choices=list(models.keys()),
|
| 85 |
-
value="
|
| 86 |
label="Select Model"
|
| 87 |
)
|
| 88 |
|
| 89 |
chat_history = gr.Chatbot(label="Chat History", elem_classes="chatbox")
|
| 90 |
|
| 91 |
user_message = gr.Textbox(
|
| 92 |
-
placeholder="Type your message here...",
|
| 93 |
show_label=False,
|
| 94 |
elem_classes="chatbox",
|
| 95 |
)
|
|
@@ -101,13 +101,11 @@ def create_chat_interface():
|
|
| 101 |
|
| 102 |
model_choice.change(fn=toggle_pdf_input, inputs=model_choice, outputs=file_input)
|
| 103 |
|
| 104 |
-
|
| 105 |
-
|
| 106 |
-
# Link the send button to the chat function
|
| 107 |
-
send_button.click(
|
| 108 |
chat_with_model,
|
| 109 |
inputs=[model_choice, user_message, chat_history, file_input],
|
| 110 |
-
outputs=chat_history,
|
| 111 |
)
|
| 112 |
|
| 113 |
return interface
|
|
|
|
| 4 |
|
| 5 |
# Models Setup
|
| 6 |
models = {
|
| 7 |
+
"Job Consultant (Zephyr)": {
|
| 8 |
"client": InferenceClient(model="HuggingFaceH4/zephyr-7b-beta"),
|
| 9 |
},
|
| 10 |
"PDF Summarizer (T5)": {
|
|
|
|
| 22 |
user_message += f"\n\nPDF Content:\n{pdf_text}"
|
| 23 |
|
| 24 |
if not user_message.strip():
|
| 25 |
+
return chat_history, ""
|
| 26 |
|
| 27 |
model_info = models[model_choice]
|
| 28 |
client = model_info["client"]
|
|
|
|
| 52 |
|
| 53 |
# Update Chat History
|
| 54 |
chat_history.append((user_message, response))
|
| 55 |
+
return chat_history, ""
|
| 56 |
|
| 57 |
# Function to Extract Text from PDF
|
| 58 |
def extract_text_from_pdf(file):
|
|
|
|
| 82 |
with gr.Row():
|
| 83 |
model_choice = gr.Dropdown(
|
| 84 |
choices=list(models.keys()),
|
| 85 |
+
value="Job Consultant (Zephyr)",
|
| 86 |
label="Select Model"
|
| 87 |
)
|
| 88 |
|
| 89 |
chat_history = gr.Chatbot(label="Chat History", elem_classes="chatbox")
|
| 90 |
|
| 91 |
user_message = gr.Textbox(
|
| 92 |
+
placeholder="Type your message here and press Enter...",
|
| 93 |
show_label=False,
|
| 94 |
elem_classes="chatbox",
|
| 95 |
)
|
|
|
|
| 101 |
|
| 102 |
model_choice.change(fn=toggle_pdf_input, inputs=model_choice, outputs=file_input)
|
| 103 |
|
| 104 |
+
# Link the input box to send messages on Enter
|
| 105 |
+
user_message.submit(
|
|
|
|
|
|
|
| 106 |
chat_with_model,
|
| 107 |
inputs=[model_choice, user_message, chat_history, file_input],
|
| 108 |
+
outputs=[chat_history, user_message],
|
| 109 |
)
|
| 110 |
|
| 111 |
return interface
|