Gradio app summary

download_judgements.py

@@ -79,3 +79,8 @@ def fetch_judgements():
     logging.info(f"returned case objs {len(case_objs)}")
 
     return case_objs
+
+def fetch_case_names():
+    case_objs = fetch_judgements()
+    names = [case['title'] for case in case_objs ]
+    return names

interpret_judgement.py

@@ -36,10 +36,10 @@ def download_dataset(url, save_folder="datasets", file_name=None):
         return f"Failed to download dataset. Status code: {response.status_code}"
 
 
-def interpret_judgement(case_url, file_name, title):
+def interpret_judgement(case_url, file_name='test', title=''):
     download_dataset(url=case_url, file_name=f"{file_name}.pdf")
     documents = SimpleDirectoryReader("datasets").load_data()
     index = VectorStoreIndex.from_documents(documents)
     query_engine = index.as_query_engine()
-    return query_engine.query(f'''Intepret and simply the judegement {title} from given context. Convert it into a 3-4 line NEWS story format.
+    return query_engine.query(f'''Intepret and simplify the judegement {title} from given context. Convert it into a 3-4 line NEWS story format.
                               It should include information about petitioner, respondent and quick summary of order passed.''')

main.py

@@ -4,15 +4,33 @@ from slack_bolt import App
 from slack_bolt.adapter.fastapi import SlackRequestHandler
 import os
 import logging
+from download_judgements import fetch_case_names
+from interpret_judgement import interpret_judgement
+
 app = FastAPI()
 
-io = gr.Interface(lambda x: "Hello, " + x + "!", "textbox", "textbox")
+# io = gr.Interface(lambda x: "Hello, " + x + "!", "textbox", "textbox")
+
+io = gr.Interface(
+    fn=interpret_judgement,
+    inputs=[
+        gr.Dropdown(
+            fetch_case_names(), label="Chose a case", info="Chose a case from latest Supreme Court judgements"
+        ),
+    ],
+    outputs=gr.outputs.Textbox(label="Summary"),
+    title="Judgement Buzz",
+    description="Your AI helper to interpret complex judgement from Supreme Court of India",
+)
+
 app = gr.mount_gradio_app(app, io, path='/')
 
-slack_app = App(token=os.environ.get("SLACK_BOT_TOKEN"), signing_secret=os.environ.get("SLACK_SIGNING_SECRET"))
+slack_app = App(token=os.environ.get("SLACK_BOT_TOKEN"),
+                signing_secret=os.environ.get("SLACK_SIGNING_SECRET"))
 app_handler = SlackRequestHandler(slack_app)
 
+
 @app.post("/slack/events")
 async def endpoint(req: Request):
     logging.info('recevied slack event')
-    return await app_handler.handle(req)
+    return await app_handler.handle(req)