Spaces:
Sleeping
Sleeping
moving around app file
Browse files- Dockerfile +2 -1
- rag_app/app.py +0 -77
- rag_app/chat_utils.py +1 -1
Dockerfile
CHANGED
|
@@ -34,4 +34,5 @@ COPY . /finance-rag-chatbot-group39
|
|
| 34 |
WORKDIR /finance-rag-chatbot-group39
|
| 35 |
|
| 36 |
# Final command: run the mesop script
|
| 37 |
-
CMD ["mesop", "rag_app/app.py", "--port", "8080"]
|
|
|
|
|
|
| 34 |
WORKDIR /finance-rag-chatbot-group39
|
| 35 |
|
| 36 |
# Final command: run the mesop script
|
| 37 |
+
# CMD ["mesop", "rag_app/app.py", "--port", "8080"]
|
| 38 |
+
CMD ["gunicorn", "--bind", "0.0.0.0:8080", "app:me"]
|
rag_app/app.py
DELETED
|
@@ -1,77 +0,0 @@
|
|
| 1 |
-
import mesop as me
|
| 2 |
-
from chat_utils import State, _make_style_chat_bubble_wrapper, _ROLE_ASSISTANT, on_chat_input, _make_chat_bubble_style, \
|
| 3 |
-
on_click_submit_chat_msg, _STYLE_CHAT_BUBBLE_NAME, handle_pdf_upload
|
| 4 |
-
|
| 5 |
-
_COLOR_BACKGROUND = me.theme_var("background")
|
| 6 |
-
|
| 7 |
-
_STYLE_APP_CONTAINER = me.Style(
|
| 8 |
-
background=_COLOR_BACKGROUND,
|
| 9 |
-
display="flex",
|
| 10 |
-
flex_direction="column",
|
| 11 |
-
height="100%",
|
| 12 |
-
margin=me.Margin.symmetric(vertical=0, horizontal="auto"),
|
| 13 |
-
width="min(1024px, 100%)",
|
| 14 |
-
box_shadow="0 3px 1px -2px #0003, 0 2px 2px #00000024, 0 1px 5px #0000001f",
|
| 15 |
-
padding=me.Padding(top=20, left=20, right=20),
|
| 16 |
-
)
|
| 17 |
-
|
| 18 |
-
|
| 19 |
-
@me.page()
|
| 20 |
-
def app():
|
| 21 |
-
state = me.state(State)
|
| 22 |
-
with me.box(style=_STYLE_APP_CONTAINER):
|
| 23 |
-
with me.box(style=me.Style(
|
| 24 |
-
width="min(680%, 100%)",
|
| 25 |
-
margin=me.Margin.symmetric(vertical=36, horizontal="auto"),
|
| 26 |
-
flex_grow=1,
|
| 27 |
-
overflow_y="auto",
|
| 28 |
-
padding=me.Padding(left=20, right=20)
|
| 29 |
-
)):
|
| 30 |
-
me.text("""
|
| 31 |
-
FinanceGPT - Powered by open source language models capable of document QnA on Annual
|
| 32 |
-
Investor Reports of top companies.
|
| 33 |
-
""",
|
| 34 |
-
style=me.Style(font_size=20, margin=me.Margin(bottom=24), text_align="center")
|
| 35 |
-
)
|
| 36 |
-
me.text("ℹ️ Upload annual reports to start asking questions.",
|
| 37 |
-
style=me.Style(font_size=12, margin=me.Margin(bottom=24), text_align="center")
|
| 38 |
-
)
|
| 39 |
-
for index, msg in enumerate(state.output):
|
| 40 |
-
with me.box(style=_make_style_chat_bubble_wrapper(msg.role), key=f"msg-{index}"):
|
| 41 |
-
if msg.role == _ROLE_ASSISTANT:
|
| 42 |
-
me.text("assistant", style=_STYLE_CHAT_BUBBLE_NAME)
|
| 43 |
-
with me.box(style=_make_chat_bubble_style(msg.role)):
|
| 44 |
-
me.markdown(msg.content)
|
| 45 |
-
if state.in_progress:
|
| 46 |
-
me.progress_spinner()
|
| 47 |
-
with me.box(key="scroll-to", style=me.Style(height=250)):
|
| 48 |
-
pass
|
| 49 |
-
with me.box(style=me.Style(
|
| 50 |
-
padding=me.Padding(top=30, left=20, right=20),
|
| 51 |
-
display="flex",
|
| 52 |
-
flex_direction="row"
|
| 53 |
-
)):
|
| 54 |
-
with me.content_uploader(
|
| 55 |
-
accepted_file_types=["application/pdf"],
|
| 56 |
-
on_upload=handle_pdf_upload,
|
| 57 |
-
type="icon",
|
| 58 |
-
style=me.Style(font_weight="bold", margin=me.Margin(right=8)),
|
| 59 |
-
):
|
| 60 |
-
me.icon("attach_file")
|
| 61 |
-
|
| 62 |
-
with me.box(style=me.Style(flex_grow=1)):
|
| 63 |
-
me.input(
|
| 64 |
-
label="Enter your prompt",
|
| 65 |
-
key=f"input-{len(state.output)}",
|
| 66 |
-
on_input=on_chat_input,
|
| 67 |
-
on_enter=on_click_submit_chat_msg,
|
| 68 |
-
style=me.Style(width="100%")
|
| 69 |
-
)
|
| 70 |
-
with me.content_button(
|
| 71 |
-
color="primary",
|
| 72 |
-
type="flat",
|
| 73 |
-
disabled=state.in_progress,
|
| 74 |
-
on_click=on_click_submit_chat_msg,
|
| 75 |
-
style=me.Style(margin=me.Margin(top=8, left=8))
|
| 76 |
-
):
|
| 77 |
-
me.icon("send" if not state.in_progress else "pending")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
rag_app/chat_utils.py
CHANGED
|
@@ -3,7 +3,7 @@ import mesop as me
|
|
| 3 |
from dataclasses import dataclass, field
|
| 4 |
from typing import Callable, Generator, Literal
|
| 5 |
import time
|
| 6 |
-
from rag import extract_final_answer, answer_question
|
| 7 |
|
| 8 |
|
| 9 |
Role = Literal["user", "assistant"]
|
|
|
|
| 3 |
from dataclasses import dataclass, field
|
| 4 |
from typing import Callable, Generator, Literal
|
| 5 |
import time
|
| 6 |
+
from rag_app.rag import extract_final_answer, answer_question
|
| 7 |
|
| 8 |
|
| 9 |
Role = Literal["user", "assistant"]
|