Update app.py

app.py

@@ -21,8 +21,13 @@ def query_nvidia_api(input_text):
 
     # Check the response status
     if response.status_code == 200:
-        # Process the successful response (adjust according to actual response structure)
-        return response.json()
+        response_data = response.json()
+        # Assuming the structure is as you provided and extracting the 'content' field
+        if 'choices' in response_data and len(response_data['choices']) > 0:
+            content = response_data['choices'][0].get('message', {}).get('content', '')
+            return content
+        else:
+            return "No content found in response"
     else:
         # Handle errors or unsuccessful responses
         return f"Error: {response.text}"
@@ -30,7 +35,7 @@ def query_nvidia_api(input_text):
 # Create the Gradio interface
 interface = gr.Interface(fn=query_nvidia_api,
                          inputs=gr.Textbox(lines=7, placeholder="Enter your text here..."),
-                         outputs=gr.JSON())
+                         outputs="text")
 
 # Launch the app
 if __name__ == "__main__":