NaimaAqeel commited on
Commit
5052e4f
·
verified ·
1 Parent(s): ed7a2c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -16
app.py CHANGED
@@ -151,32 +151,29 @@ def process_and_query(state, question):
151
 
152
  # Define the Gradio interface
153
  def main():
154
- upload_tab = gr.Interface(
155
  fn=upload_files,
156
- inputs=gr.inputs.File(label="Upload PDF or DOCX files", multiple=True),
157
- outputs=gr.outputs.Text(label="Upload Status", default="No file uploaded yet", type="textbox"),
158
  live=True,
159
  capture_session=True
160
  )
161
 
162
- query_tab = gr.Interface(
163
  fn=process_and_query,
164
- inputs=gr.inputs.Textbox(label="Enter your query"),
165
- outputs=gr.outputs.Textbox(label="Query Response", default="No query processed yet", type="textbox"),
166
  live=True,
167
  capture_session=True
168
  )
169
 
170
- gr.Interface(
171
- fn=None,
172
- inputs=[
173
- gr.Interface.Tab("Upload Files", upload_tab),
174
- gr.Interface.Tab("Query", query_tab)
175
- ],
176
- outputs=gr.outputs.Textbox(label="Output", default="Output will be shown here", type="textbox"),
177
- live=True,
178
- capture_session=True
179
- ).launch()
180
 
181
  if __name__ == "__main__":
182
  main()
 
151
 
152
  # Define the Gradio interface
153
  def main():
154
+ upload_interface = gr.Interface(
155
  fn=upload_files,
156
+ inputs=gr.File(label="Upload PDF or DOCX files", multiple=True),
157
+ outputs=gr.Textbox(label="Upload Status"),
158
  live=True,
159
  capture_session=True
160
  )
161
 
162
+ query_interface = gr.Interface(
163
  fn=process_and_query,
164
+ inputs=[gr.State(), gr.Textbox(label="Enter your query")],
165
+ outputs=gr.Textbox(label="Query Response"),
166
  live=True,
167
  capture_session=True
168
  )
169
 
170
+ with gr.Blocks() as demo:
171
+ with gr.Tab("Upload Files"):
172
+ upload_interface.render()
173
+ with gr.Tab("Query"):
174
+ query_interface.render()
175
+
176
+ demo.launch()
 
 
 
177
 
178
  if __name__ == "__main__":
179
  main()