Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -38,7 +38,8 @@ def handle_chat(user_message):
|
|
| 38 |
global chat_history
|
| 39 |
try:
|
| 40 |
if not user_message.strip():
|
| 41 |
-
|
|
|
|
| 42 |
|
| 43 |
# Add user message to chat history
|
| 44 |
|
|
@@ -71,9 +72,9 @@ def handle_chat(user_message):
|
|
| 71 |
img.save(buffered, format="PNG")
|
| 72 |
img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
|
| 73 |
|
| 74 |
-
img = f'<img src="data:image/png;base64,{img_str}" style="width:
|
| 75 |
|
| 76 |
-
ai_response =
|
| 77 |
|
| 78 |
user_message = visual_query.strip()
|
| 79 |
|
|
@@ -85,7 +86,7 @@ def handle_chat(user_message):
|
|
| 85 |
return chat_history
|
| 86 |
|
| 87 |
except Exception as e:
|
| 88 |
-
chat_history.append((visual_query, f"
|
| 89 |
return chat_history
|
| 90 |
# Chat handling function
|
| 91 |
def chat_interaction(user_message):
|
|
@@ -95,12 +96,12 @@ def chat_interaction(user_message):
|
|
| 95 |
|
| 96 |
with gr.Blocks(theme="soft") as app: # Use the 'soft' theme
|
| 97 |
gr.Markdown(
|
| 98 |
-
"<h1 style='color:blue;'>
|
| 99 |
elem_id="title"
|
| 100 |
)
|
| 101 |
|
| 102 |
with gr.Tab("Chat"):
|
| 103 |
-
chat_box = gr.Chatbot(label="Chat History")
|
| 104 |
with gr.Row(): # Place elements in the same row
|
| 105 |
with gr.Column(scale=2):
|
| 106 |
user_input = gr.Textbox(
|
|
@@ -120,14 +121,14 @@ with gr.Blocks(theme="soft") as app: # Use the 'soft' theme
|
|
| 120 |
outputs=[chat_box, user_input]
|
| 121 |
)
|
| 122 |
|
| 123 |
-
|
| 124 |
upload_button = gr.UploadButton(
|
| 125 |
label="Upload your file (PDF, Excel, Docx, Txt, CSV)",
|
| 126 |
file_types=[".pdf", ".xlsx", ".docx", ".txt", ".csv"],
|
| 127 |
file_count="single"
|
| 128 |
)
|
| 129 |
file_output = gr.Textbox(label="File Processing Output")
|
| 130 |
-
upload_button.upload(fn=handle_file_uploads, inputs=upload_button, outputs=[file_output])
|
| 131 |
|
| 132 |
# Clear input field function
|
| 133 |
def clear_input():
|
|
@@ -170,6 +171,11 @@ app.css = """
|
|
| 170 |
#title {
|
| 171 |
margin-bottom: 10px; /* Space below the title */
|
| 172 |
text-align: center;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 173 |
}
|
| 174 |
"""
|
| 175 |
|
|
|
|
| 38 |
global chat_history
|
| 39 |
try:
|
| 40 |
if not user_message.strip():
|
| 41 |
+
chat_history.append(("AI: ", "Please enter a question."))
|
| 42 |
+
|
| 43 |
|
| 44 |
# Add user message to chat history
|
| 45 |
|
|
|
|
| 72 |
img.save(buffered, format="PNG")
|
| 73 |
img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
|
| 74 |
|
| 75 |
+
img = f'<img src="data:image/png;base64,{img_str}" style="width:600px; height:500px;">'
|
| 76 |
|
| 77 |
+
ai_response = img + ai_response
|
| 78 |
|
| 79 |
user_message = visual_query.strip()
|
| 80 |
|
|
|
|
| 86 |
return chat_history
|
| 87 |
|
| 88 |
except Exception as e:
|
| 89 |
+
chat_history.append((visual_query, f"As you know I am still learning at this moment I am not able to respond to your question.\nThank you for your patience!"))
|
| 90 |
return chat_history
|
| 91 |
# Chat handling function
|
| 92 |
def chat_interaction(user_message):
|
|
|
|
| 96 |
|
| 97 |
with gr.Blocks(theme="soft") as app: # Use the 'soft' theme
|
| 98 |
gr.Markdown(
|
| 99 |
+
"<h1 style='color:blue;'>Material Forecast - Generative AI Agent</h1>",
|
| 100 |
elem_id="title"
|
| 101 |
)
|
| 102 |
|
| 103 |
with gr.Tab("Chat"):
|
| 104 |
+
chat_box = gr.Chatbot(label="Chat History",elem_id="chatbot_gpt")
|
| 105 |
with gr.Row(): # Place elements in the same row
|
| 106 |
with gr.Column(scale=2):
|
| 107 |
user_input = gr.Textbox(
|
|
|
|
| 121 |
outputs=[chat_box, user_input]
|
| 122 |
)
|
| 123 |
|
| 124 |
+
"""with gr.Tab("File Upload"):
|
| 125 |
upload_button = gr.UploadButton(
|
| 126 |
label="Upload your file (PDF, Excel, Docx, Txt, CSV)",
|
| 127 |
file_types=[".pdf", ".xlsx", ".docx", ".txt", ".csv"],
|
| 128 |
file_count="single"
|
| 129 |
)
|
| 130 |
file_output = gr.Textbox(label="File Processing Output")
|
| 131 |
+
upload_button.upload(fn=handle_file_uploads, inputs=upload_button, outputs=[file_output])"""
|
| 132 |
|
| 133 |
# Clear input field function
|
| 134 |
def clear_input():
|
|
|
|
| 171 |
#title {
|
| 172 |
margin-bottom: 10px; /* Space below the title */
|
| 173 |
text-align: center;
|
| 174 |
+
|
| 175 |
+
#chatbot_gpt {
|
| 176 |
+
height: 600px !important; /* Adjust height as needed */
|
| 177 |
+
}
|
| 178 |
+
|
| 179 |
}
|
| 180 |
"""
|
| 181 |
|