Add basic structure

app.py

@@ -1,10 +1,20 @@
-import streamlit as st
+import gradio as gr
+from categories import Category
 
-st.title("Automatic Reimbursement Tool Demo")
+def predict(text):
+    pass
 
-with st.container():
-    col1, col2 = st.columns(2)
 
-    with col1:
-        st.header("Input")
-        st.file_uploader("Upload a PDF file or an image", type=["pdf", "png", "jpg", "jpeg"])
+_input = gr.File(type="binary", file_count="single", file_types=["image", "pdf"], label="Upload a receipt as a document or as an image")
+category_output = gr.Dropdown(Category.__members__.values(), value=Category.RANDOM, label="Identified category")
+information_output = gr.Json(label="Extracted information")
+
+demo = gr.Interface(
+    title="Automatic Reimbursement Tool Demo",
+    description="Description",
+    fn=predict,
+    inputs=_input,
+    outputs=[category_output, information_output],
+)
+
+demo.launch(debug=True, show_error=True)

categories/__init__.py

@@ -1,22 +1,17 @@
 from enum import Enum
-
-from . import random_
-from . import accomodation
-from . import travel_cab
-from . import travel_flight
+from typing import Union
 
 # from . import vendor
 from langchain.chains import LLMChain
 from langchain.chat_models import ChatOpenAI
 from langchain.output_parsers import PydanticOutputParser
 from langchain.output_parsers.enum import EnumOutputParser
-from langchain.prompts import (
-    ChatPromptTemplate,
-    HumanMessagePromptTemplate,
-    SystemMessagePromptTemplate,
-)
+from langchain.prompts import (ChatPromptTemplate, HumanMessagePromptTemplate,
+                               SystemMessagePromptTemplate)
 from pydantic import BaseModel
 
+from . import accomodation, random_, travel_cab, travel_flight
+
 
 class Category(Enum):
     ACCOMODATION = "accomodation"
@@ -83,7 +78,7 @@ def categorize_text(text: str) -> Category:
     )
 
 
-def run_category_chain(category: Category, text: str) -> BaseModel | None:
+def run_category_chain(category: Category, text: str) -> Union[BaseModel, None]:
     """Runs the chain for the given category on the given text.
 
     Args:

categories/random_/model.py

@@ -48,7 +48,7 @@ class SellerDetails(BaseModel):
     )
 
 
-class UIDs(BaseModel):
+class UIDDict(BaseModel):
     invoice_number: str = Field(..., title="The invoice number")
     other_uids: Dict[str, str] = Field(
         ...,
@@ -57,7 +57,7 @@ class UIDs(BaseModel):
 
 
 class InformationExtractedFromABillReceipt(BaseModel):
-    uids: UIDs = Field(..., title="Invoice number and other UIDs")
+    uids: UIDDict = Field(..., title="Invoice number and other UIDs")
     total: float = Field(..., title="Total amount or price")
     tax: Union[TaxItem, TaxItem1] = Field(..., title="The total tax amount")
     name: str = Field(
@@ -68,7 +68,7 @@ class InformationExtractedFromABillReceipt(BaseModel):
         default="INR",
         title="The ISO 4217 code for the currency in which the prices in the invoice are (inferred from symbols, names, addresses, etc)",
     )
-    date: date = Field(
+    issue_date: date = Field(
         ..., title="The date the invoice was issued"
     )
     seller_details: SellerDetails = Field(..., title="Information about the seller")

requirements.txt

@@ -23,7 +23,6 @@ backcall==0.2.0
 bcrypt==3.2.0
 beautifulsoup4==4.12.2
 binaryornot==0.4.4
-black==0.0
 bleach==4.1.0
 bokeh==3.1.1
 botocore==1.29.76